Searched refs:kCudnnBatchNormBackwardCallTarget (Results 1 – 6 of 6) sorted by relevance
342 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateOperandConstraint()387 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateBufferConstraint()
96 extern const char* const kCudnnBatchNormBackwardCallTarget;
276 batch_norm_shape, operands, kCudnnBatchNormBackwardCallTarget)); in HandleBatchNormGrad()
168 const char* const kCudnnBatchNormBackwardCallTarget = variable178 target == kCudnnBatchNormBackwardCallTarget; in IsCustomCallToDnnBatchNorm()
301 kCudnnBatchNormBackwardCallTarget)); in TEST_F()
874 if (target == xla::gpu::kCudnnBatchNormBackwardCallTarget) { in EmitDnnBatchNorm()