| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNorm.pbtxt | 36 name: "reserve_space_1"
115 name: "reserve_space_1"
|
| D | FusedBatchNormV2.pbtxt | 36 name: "reserve_space_1"
126 name: "reserve_space_1"
|
| D | FusedBatchNormV3.pbtxt | 36 name: "reserve_space_1"
130 name: "reserve_space_1"
|
| D | FusedBatchNormGrad.pbtxt | 16 name: "reserve_space_1"
|
| D | FusedBatchNormGradV3.pbtxt | 16 name: "reserve_space_1"
|
| D | FusedBatchNormGradV2.pbtxt | 16 name: "reserve_space_1"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNorm.pbtxt | 36 name: "reserve_space_1"
115 name: "reserve_space_1"
|
| D | FusedBatchNormV2.pbtxt | 36 name: "reserve_space_1"
126 name: "reserve_space_1"
|
| D | FusedBatchNormGradV3.pbtxt | 16 name: "reserve_space_1"
110 name: "reserve_space_1"
|
| D | FusedBatchNormV3.pbtxt | 36 name: "reserve_space_1"
137 name: "reserve_space_1"
|
| D | FusedBatchNormGrad.pbtxt | 16 name: "reserve_space_1"
|
| D | FusedBatchNormGradV2.pbtxt | 16 name: "reserve_space_1"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 134 {fused_batch_norm_tf.reserve_space_1.name(), in TEST()
135 fused_batch_norm_xla.reserve_space_1.name(), in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 54 Tensor reserve_space_1; member
197 fwd.reserve_space_1, fwd.reserve_space_2, fwd.reserve_space_3, in RunFusedBatchNorm()
214 forward->reserve_space_1 = out_tensors[3]; in RunFusedBatchNorm()
326 forward->reserve_space_1 = out_tensors[3]; in RunFusedBatchNormEx()
400 {fbn_forward.reserve_space_1, fbn_ex_forward.reserve_space_1}, in VerifyTensorsNear()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 261 reserve_space_1 = array_ops.placeholder(
270 reserve_space_1,
279 reserve_space_1: reserve_space_1_val,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNorm.pbtxt | 56 name: "reserve_space_1"
|
| D | api_def_FusedBatchNormGrad.pbtxt | 22 name: "reserve_space_1"
|
| D | api_def_FusedBatchNormV2.pbtxt | 56 name: "reserve_space_1"
|
| D | api_def_FusedBatchNormV3.pbtxt | 56 name: "reserve_space_1"
|
| D | api_def_FusedBatchNormGradV2.pbtxt | 22 name: "reserve_space_1"
|
| D | api_def_FusedBatchNormGradV3.pbtxt | 22 name: "reserve_space_1"
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1021 DeviceMemory<float>* reserve_space_1, in DoBatchNormalizationForward() argument
1040 DeviceMemory<float>* reserve_space_1, in DoBatchNormalizationForward() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 69 /*reserve_space_1=*/(replaceWithValue $x),
|
| D | legalize_tf.cc | 1812 Value mean = op.reserve_space_1(); in matchAndRewrite()
1988 Value reserve_space_1 = batch_mean; in matchAndRewrite() local
2052 /*reserve_space_1=*/reserve_space_1, in matchAndRewrite()
2071 /*reserve_space_1=*/reserve_space_1, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 264 auto reserve_space_1 = ops::RandomUniform(scope.WithOpName("reserve_space_1"), in CreateSimpleFusedBatchNormGrad() local
270 y_backprop, x, scale, reserve_space_1, reserve_space_2, in CreateSimpleFusedBatchNormGrad()
|