Home
last modified time | relevance | path

Searched refs:kCudnnBatchNormForwardTrainingCallTarget (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc328 kCudnnBatchNormForwardTrainingCallTarget && in PropagateOperandConstraint()
381 kCudnnBatchNormForwardTrainingCallTarget && in PropagateBufferConstraint()
Dir_emission_utils.h95 extern const char* const kCudnnBatchNormForwardTrainingCallTarget;
Dcudnn_batchnorm_rewriter.cc166 kCudnnBatchNormForwardTrainingCallTarget)); in HandleBatchNormTraining()
Dnvptx_compiler.cc147 kCudnnBatchNormForwardTrainingCallTarget); in OptimizeHloConvolutionCanonicalization()
Dir_emission_utils.cc166 const char* const kCudnnBatchNormForwardTrainingCallTarget = variable
177 target == kCudnnBatchNormForwardTrainingCallTarget || in IsCustomCallToDnnBatchNorm()
Dgpu_layout_assignment_test.cc220 kCudnnBatchNormForwardTrainingCallTarget)); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc867 if (target == xla::gpu::kCudnnBatchNormForwardTrainingCallTarget) { in EmitDnnBatchNorm()