Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h84 extern const char* const kCudnnBatchNormBackwardCallTarget;
Dir_emission_utils.cc113 const char* const kCudnnBatchNormBackwardCallTarget = variable
123 target == kCudnnBatchNormBackwardCallTarget; in IsCustomCallToDnnBatchNorm()
Dgpu_layout_assignment.cc309 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateOperandConstraint()
354 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateBufferConstraint()
Dcudnn_batchnorm_rewriter.cc201 batch_norm->shape(), operands, kCudnnBatchNormBackwardCallTarget); in HandleBatchNormGrad()
Dcudnn_batchnorm_thunk.cc236 CHECK_EQ(hlo->custom_call_target(), kCudnnBatchNormBackwardCallTarget); in CudnnBatchNormBackwardThunk()
Dgpu_layout_assignment_test.cc299 kCudnnBatchNormBackwardCallTarget)); in TEST_F()
Dir_emitter_unnested.cc435 if (custom_call->custom_call_target() == kCudnnBatchNormBackwardCallTarget) { in HandleCustomCall()