Home
last modified time | relevance | path

Searched defs:reserve_space_1 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_ex_op_test.cc54 Tensor reserve_space_1; member
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1021 DeviceMemory<float>* reserve_space_1, in DoBatchNormalizationForward()
1040 DeviceMemory<float>* reserve_space_1, in DoBatchNormalizationForward()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc264 auto reserve_space_1 = ops::RandomUniform(scope.WithOpName("reserve_space_1"), in CreateSimpleFusedBatchNormGrad() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1988 Value reserve_space_1 = batch_mean; in matchAndRewrite() local