Home
last modified time | relevance | path

Searched defs:reserve_space_2 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_ex_op_test.cc55 Tensor reserve_space_2; member
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1022 DeviceMemory<float>* reserve_space_2, bool is_training, in DoBatchNormalizationForward()
1041 DeviceMemory<float>* reserve_space_2, bool is_training, in DoBatchNormalizationForward()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc266 auto reserve_space_2 = ops::RandomUniform(scope.WithOpName("reserve_space_2"), in CreateSimpleFusedBatchNormGrad() local