| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 83 using ::tensorflow::ops::FusedBatchNorm; in TEST()
114 FusedBatchNorm fused_batch_norm_tf( in TEST()
116 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
117 FusedBatchNorm fused_batch_norm_xla( in TEST()
119 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNorm.pbtxt | 2 name: "FusedBatchNorm"
81 name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNorm.pbtxt | 2 name: "FusedBatchNorm"
81 name: "FusedBatchNorm"
|
| /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/compiler/jit/tests/ |
| D | keras_imagenet_main_graph_mode.pbtxt | 29718 name: "bn_conv1_1/FusedBatchNorm"
29719 op: "FusedBatchNorm"
29827 input: "bn_conv1_1/FusedBatchNorm:1"
29848 input: "bn_conv1_1/FusedBatchNorm:2"
29868 input: "bn_conv1_1/FusedBatchNorm"
30227 name: "bn2a_branch1_1/FusedBatchNorm"
30228 op: "FusedBatchNorm"
30261 name: "bn2a_branch2a_1/FusedBatchNorm"
30262 op: "FusedBatchNorm"
30298 input: "bn2a_branch1_1/FusedBatchNorm:1"
[all …]
|
| D | keras_imagenet_main_graph_mode.golden_summary | 29 FusedBatchNorm 53
|
| /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_FusedBatchNormGradV3.pbtxt | 69 Unused placeholder to match the mean input in FusedBatchNorm.
76 in FusedBatchNorm.
|
| D | api_def_FusedBatchNorm.pbtxt | 2 graph_op_name: "FusedBatchNorm"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 44 TEST_F(RemapperTest, FusedBatchNorm) { in TEST_F() argument
52 ops::FusedBatchNorm::Attrs attr; in TEST_F()
54 ops::FusedBatchNorm bn(s.WithOpName("batch_norm"), x, scale, offset, mean, in TEST_F()
88 ops::FusedBatchNorm::Attrs attr; in TEST_F()
91 ops::FusedBatchNorm bn(s.WithOpName("batch_norm").WithDevice("/device:GPU:0"), in TEST_F()
773 ops::FusedBatchNorm::Attrs attrs; in TEST_F()
775 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, scale, in TEST_F()
852 ops::FusedBatchNorm::Attrs attrs; in TEST_F()
854 auto batch_norm = ops::FusedBatchNorm(s.WithOpName("batch_norm"), conv, in TEST_F()
|
| D | remapper.cc | 92 struct FusedBatchNorm { struct
93 FusedBatchNorm() = default;
94 explicit FusedBatchNorm(int fused_batch_norm) in FusedBatchNorm() function
773 FusedBatchNorm* matched) { in IsCpuCompatibleDataType()
1444 Status AddBatchNormNodes(RemapperContext* ctx, const FusedBatchNorm& matched) { in IsCpuCompatibleDataType()
1921 FusedBatchNorm fused_batch_norm; in IsCpuCompatibleDataType()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 681 ops::FusedBatchNorm::Attrs attr; in RunConv2DWithBatchNorm()
684 ops::FusedBatchNorm with_fused_batch_norm = ops::FusedBatchNorm( in RunConv2DWithBatchNorm()
711 ops::FusedBatchNorm::Attrs attr; in RunConv2DWithBatchNormAndActivation()
714 ops::FusedBatchNorm with_fused_batch_norm = ops::FusedBatchNorm( in RunConv2DWithBatchNormAndActivation()
|
| D | fused_batch_norm_op_test.cc | 289 …static void BM_NAME(FusedBatchNorm, N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE)(::testing::benchma…
298 BM_NAME(FusedBatchNorm, N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE)) \
|
| D | fused_batch_norm_op.cc | 85 struct FusedBatchNorm;
92 struct FusedBatchNorm<CPUDevice, T, U, /* is_training= */ true> { struct
230 struct FusedBatchNorm<CPUDevice, T, U, /* is_training= */ false> { struct
758 struct FusedBatchNorm<GPUDevice, T, U, is_training> { struct
1318 functor::FusedBatchNorm<Device, T, U, true>()( in ComputeWithReservedSpace()
1324 functor::FusedBatchNorm<Device, T, U, false>()( in ComputeWithReservedSpace()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/ |
| D | v1_1.0_224_frozen.wrong_attr.line.part.pbtxt | 1 …4,224,3 -tf-output-arrays=MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/FusedBatchNorm -tf-debug-info…
311 name: "MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/FusedBatchNorm"
312 op: "FusedBatchNorm"
|
| D | v1_1.0_224_frozen.wrong_attr.stack.part.pbtxt | 1 …4,224,3 -tf-output-arrays=MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/FusedBatchNorm -tf-debug-info…
314 name: "MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/FusedBatchNorm"
315 op: "FusedBatchNorm"
|
| D | v1_1.0_224_frozen.wrong_attr.stack.part.pbtxt.debug | 150 key: "MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/FusedBatchNorm@"
|
| D | v1_1.0_224_frozen.wrong_attr.line.part.pbtxt.debug | 150 key: "MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/FusedBatchNorm@"
|
| /external/tensorflow/tensorflow/core/profiler/g3doc/ |
| D | profile_memory.md | 28 FusedBatchNorm 10517.52MB (64.01%, 8.74%)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | gpu_fusion.mlir | 40 // CHECK-NEXT: %[[Y:[a-z0-9]*]], {{.*}}FusedBatchNorm
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 244 ops::FusedBatchNorm::Attrs attr; in VerifyFusedBatchNorm()
248 auto bn = ops::FusedBatchNorm(root.WithOpName("FusedBatchNorm"), input_op, in VerifyFusedBatchNorm()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 886 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithBatchNorm()
888 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithBatchNorm()
2223 auto batch_norm = ops::FusedBatchNorm( in CreateGrapplerItemWithInterDeviceTransfers()
2225 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithInterDeviceTransfers()
|