aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/receptive_field
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-01-09 11:12:26 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-09 11:16:03 -0800
commite95ca59ecd421a1fcda16ca83372df2a5908b179 (patch)
tree77fa0189dbd98b14408ca00054a9178e58f30c79 /tensorflow/contrib/receptive_field
parentc4889755e0ea8dca3d8b14d7dfce7560e545fc2f (diff)
Support FusedBatchNorm layer in receptive field calculation.
PiperOrigin-RevId: 181345319
Diffstat (limited to 'tensorflow/contrib/receptive_field')
-rw-r--r--tensorflow/contrib/receptive_field/python/util/examples/rf_benchmark.py12
-rw-r--r--tensorflow/contrib/receptive_field/python/util/examples/write_inception_resnet_v2_graph.py2
-rw-r--r--tensorflow/contrib/receptive_field/python/util/receptive_field.py6
3 files changed, 10 insertions, 10 deletions
diff --git a/tensorflow/contrib/receptive_field/python/util/examples/rf_benchmark.py b/tensorflow/contrib/receptive_field/python/util/examples/rf_benchmark.py
index 94228dfa61..5db8dacdc8 100644
--- a/tensorflow/contrib/receptive_field/python/util/examples/rf_benchmark.py
+++ b/tensorflow/contrib/receptive_field/python/util/examples/rf_benchmark.py
@@ -28,12 +28,6 @@ import argparse
import csv
import sys
-from nets import alexnet
-from nets import inception
-from nets import mobilenet_v1
-from nets import resnet_v1
-from nets import resnet_v2
-from nets import vgg
from tensorflow.contrib import framework
from tensorflow.contrib import receptive_field
from tensorflow.contrib import slim
@@ -41,6 +35,12 @@ from tensorflow.python.framework import dtypes
from tensorflow.python.framework import ops
from tensorflow.python.ops import array_ops
from tensorflow.python.platform import app
+from nets import alexnet
+from nets import inception
+from nets import mobilenet_v1
+from nets import resnet_v1
+from nets import resnet_v2
+from nets import vgg
cmd_args = None
diff --git a/tensorflow/contrib/receptive_field/python/util/examples/write_inception_resnet_v2_graph.py b/tensorflow/contrib/receptive_field/python/util/examples/write_inception_resnet_v2_graph.py
index 793ae163d8..a494883396 100644
--- a/tensorflow/contrib/receptive_field/python/util/examples/write_inception_resnet_v2_graph.py
+++ b/tensorflow/contrib/receptive_field/python/util/examples/write_inception_resnet_v2_graph.py
@@ -22,12 +22,12 @@ from __future__ import print_function
import argparse
import sys
-from nets import inception
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import graph_io
from tensorflow.python.framework import ops
from tensorflow.python.ops import array_ops
from tensorflow.python.platform import app
+from nets import inception
cmd_args = None
diff --git a/tensorflow/contrib/receptive_field/python/util/receptive_field.py b/tensorflow/contrib/receptive_field/python/util/receptive_field.py
index 6d207347d8..0955e1fc39 100644
--- a/tensorflow/contrib/receptive_field/python/util/receptive_field.py
+++ b/tensorflow/contrib/receptive_field/python/util/receptive_field.py
@@ -33,9 +33,9 @@ import numpy as np
# White-listed layer operations, which do not affect the receptive field
# computation.
_UNCHANGED_RF_LAYER_OPS = [
- "Add", "BiasAdd", "Cast", "Ceil", "ConcatV2", "Const", "Floor", "Identity",
- "Log", "Mul", "Pow", "RealDiv", "Relu", "Relu6", "Round", "Rsqrt",
- "Softplus", "Sub", "VariableV2"
+ "Add", "BiasAdd", "Cast", "Ceil", "ConcatV2", "Const", "Floor",
+ "FusedBatchNorm", "Identity", "Log", "Mul", "Pow", "RealDiv", "Relu",
+ "Relu6", "Round", "Rsqrt", "Softplus", "Sub", "VariableV2"
]
# Different ways in which padding modes may be spelled.