diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-09 11:12:26 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-09 11:16:03 -0800 |
| commit | e95ca59ecd421a1fcda16ca83372df2a5908b179 (patch) | |
| tree | 77fa0189dbd98b14408ca00054a9178e58f30c79 /tensorflow/contrib/receptive_field | |
| parent | c4889755e0ea8dca3d8b14d7dfce7560e545fc2f (diff) | |
Support FusedBatchNorm layer in receptive field calculation.
PiperOrigin-RevId: 181345319
Diffstat (limited to 'tensorflow/contrib/receptive_field')
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. |