| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNorm.pbtxt | 40 name: "reserve_space_2"
119 name: "reserve_space_2"
|
| D | FusedBatchNormV2.pbtxt | 40 name: "reserve_space_2"
130 name: "reserve_space_2"
|
| D | FusedBatchNormV3.pbtxt | 40 name: "reserve_space_2"
134 name: "reserve_space_2"
|
| D | FusedBatchNormGrad.pbtxt | 20 name: "reserve_space_2"
|
| D | FusedBatchNormGradV3.pbtxt | 20 name: "reserve_space_2"
|
| D | FusedBatchNormGradV2.pbtxt | 20 name: "reserve_space_2"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNorm.pbtxt | 40 name: "reserve_space_2"
119 name: "reserve_space_2"
|
| D | FusedBatchNormV2.pbtxt | 40 name: "reserve_space_2"
130 name: "reserve_space_2"
|
| D | FusedBatchNormGradV3.pbtxt | 20 name: "reserve_space_2"
114 name: "reserve_space_2"
|
| D | FusedBatchNormV3.pbtxt | 40 name: "reserve_space_2"
141 name: "reserve_space_2"
|
| D | FusedBatchNormGrad.pbtxt | 20 name: "reserve_space_2"
|
| D | FusedBatchNormGradV2.pbtxt | 20 name: "reserve_space_2"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 136 fused_batch_norm_tf.reserve_space_2.name(), in TEST()
137 fused_batch_norm_xla.reserve_space_2.name()}, in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 55 Tensor reserve_space_2; member
197 fwd.reserve_space_1, fwd.reserve_space_2, fwd.reserve_space_3, in RunFusedBatchNorm()
215 forward->reserve_space_2 = out_tensors[4]; in RunFusedBatchNorm()
327 forward->reserve_space_2 = out_tensors[4]; in RunFusedBatchNormEx()
401 {fbn_forward.reserve_space_2, fbn_ex_forward.reserve_space_2}, in VerifyTensorsNear()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 263 reserve_space_2 = array_ops.placeholder(
271 reserve_space_2,
280 reserve_space_2: reserve_space_2_val,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNorm.pbtxt | 63 name: "reserve_space_2"
|
| D | api_def_FusedBatchNormGrad.pbtxt | 31 name: "reserve_space_2"
|
| D | api_def_FusedBatchNormV2.pbtxt | 63 name: "reserve_space_2"
|
| D | api_def_FusedBatchNormV3.pbtxt | 63 name: "reserve_space_2"
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 31 name: "reserve_space_2"
|
| D | api_def_FusedBatchNormGradV3.pbtxt | 31 name: "reserve_space_2"
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1022 DeviceMemory<float>* reserve_space_2, bool is_training, in DoBatchNormalizationForward() argument
1041 DeviceMemory<float>* reserve_space_2, bool is_training, in DoBatchNormalizationForward() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 70 /*reserve_space_2=*/(replaceWithValue $x)],
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 266 auto reserve_space_2 = ops::RandomUniform(scope.WithOpName("reserve_space_2"), in CreateSimpleFusedBatchNormGrad() local
270 y_backprop, x, scale, reserve_space_1, reserve_space_2, in CreateSimpleFusedBatchNormGrad()
|
| D | auto_mixed_precision_test.cc | 519 Output fbn1_rs2 = fbn1_op.reserve_space_2; in TEST_F()
|