Home
last modified time | relevance | path

Searched refs:output_mean (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc78 const BufferAllocation::Slice& output_mean, in CudnnBatchNormForwardTrainingThunk() argument
86 output_mean_(output_mean), in CudnnBatchNormForwardTrainingThunk()
96 se::DeviceMemory<float> output_mean( in ExecuteOnStream() local
106 config_, operand, output_data, output_mean, output_inv_stddev, in ExecuteOnStream()
Dcudnn_batchnorm_runner.cc51 se::DeviceMemory<float> output_mean; member
182 /*saved_mean=*/&params->output_mean, // in RunCudnnBatchNormForwardTrainingImpl()
257 se::DeviceMemoryBase output_data, se::DeviceMemory<float> output_mean, in RunCudnnBatchNormForwardTraining() argument
264 forward_params.output_mean = output_mean; in RunCudnnBatchNormForwardTraining()
Dcudnn_batchnorm_runner.h50 se::DeviceMemoryBase output_data, se::DeviceMemory<float> output_mean,
Dcudnn_batchnorm_thunk.h84 const BufferAllocation::Slice& output_mean,
Dir_emitter_unnested.cc1373 TF_ASSIGN_OR_RETURN(auto output_mean, in EmitBatchNormThunkFromMlir()
1386 /*output_mean=*/output_mean, in EmitBatchNormThunkFromMlir()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc145 auto output_mean = in CreateSimpleFusedBatchNorm() local
816 auto* output_mean = context.graph_view->GetNode("output_mean"); in TEST_F() local
817 ASSERT_NE(output_mean, nullptr); in TEST_F()
818 ASSERT_EQ(output_mean->NumRegularFanins(), 1); in TEST_F()
819 VerifyRegularFaninMatch(output_mean, 0, bn_node->GetName(), 1); in TEST_F()