| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 69 using ::tensorflow::ops::FusedBatchNorm; in TEST()
100 FusedBatchNorm fused_batch_norm_tf( in TEST()
102 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
103 FusedBatchNorm fused_batch_norm_xla( in TEST()
105 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 39 TEST_F(RemapperTest, FusedBatchNorm) { in TEST_F() argument
47 ops::FusedBatchNorm::Attrs attr; in TEST_F()
49 ops::FusedBatchNorm bn(s.WithOpName("batch_norm"), x, scale, offset, mean, in TEST_F()
80 ops::FusedBatchNorm::Attrs attr; in TEST_F()
83 ops::FusedBatchNorm bn(s.WithOpName("batch_norm").WithDevice("/device:GPU:0"), in TEST_F()
252 ops::FusedBatchNorm::Attrs attrs; in TEST_F()
254 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F()
329 ops::FusedBatchNorm::Attrs attrs; in TEST_F()
331 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F()
|
| D | remapper.cc | 63 struct FusedBatchNorm { struct
64 FusedBatchNorm() = default;
65 explicit FusedBatchNorm(const NodeDef* fused_batch_norm) in FusedBatchNorm() function
434 FusedBatchNorm* matched) { in FindFusedBatchNorm()
631 void AddBatchNormNodes(const FusedBatchNorm& matched, in AddBatchNormNodes()
779 FusedBatchNorm fused_batch_norm; in Optimize()
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_FusedBatchNorm.pbtxt | 2 graph_op_name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_FusedBatchNorm.pbtxt | 2 graph_op_name: "FusedBatchNorm"
|
| D | api_def_FusedBatchNormV2.pbtxt | 4 name: "nn.FusedBatchNorm"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormGrad.pbtxt | 61 Unused placeholder to match the mean input in FusedBatchNorm.
68 in FusedBatchNorm.
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 61 Unused placeholder to match the mean input in FusedBatchNorm.
68 in FusedBatchNorm.
|
| D | api_def_FusedBatchNorm.pbtxt | 2 graph_op_name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 41 struct FusedBatchNorm;
66 struct FusedBatchNorm<CPUDevice, T, U> { struct
241 struct FusedBatchNorm<GPUDevice, T, U> { struct
587 functor::FusedBatchNorm<Device, T, U>()( in Compute()
|
| D | conv_ops_test.cc | 663 ops::FusedBatchNorm::Attrs attr; in RunConv2DWithBatchNorm()
666 ops::FusedBatchNorm with_fused_batch_norm = ops::FusedBatchNorm( in RunConv2DWithBatchNorm()
690 ops::FusedBatchNorm::Attrs attr; in RunConv2DWithBatchNormAndRelu()
693 ops::FusedBatchNorm with_fused_batch_norm = ops::FusedBatchNorm( in RunConv2DWithBatchNormAndRelu()
|
| /external/tensorflow/tensorflow/core/profiler/g3doc/ |
| D | profile_memory.md | 28 FusedBatchNorm 10517.52MB (64.01%, 8.74%)
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 269 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm()
271 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithBatchNorm()
1557 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers()
1559 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithInterDeviceTransfers()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 89 `FusedBatchNorm` | `T={float}`
|
| D | cpu_supported_ops.md | 89 `FusedBatchNorm` | `T={float}`
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | README.md | 155 with a single op like BatchNormWithGlobalNormalization or FusedBatchNorm, and
410 `FusedBatchNorm`. In modern versions, adding batch normalization from Python
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1348 name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1348 name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 10719 func FusedBatchNorm(scope *Scope, x tf.Output, scale tf.Output, offset tf.Output, mean tf.Output, v… func
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 12750 name: "FusedBatchNorm"
|
| D | ops_history.v1.pbtxt | 27700 name: "FusedBatchNorm"
27773 name: "FusedBatchNorm"
|
| D | ops_history.v2.pbtxt | 27329 name: "FusedBatchNorm"
27402 name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13713 name: "FusedBatchNorm"
|