Searched refs:kCudnnBatchNormForwardInferenceCallTarget (Results 1 – 6 of 6) sorted by relevance
316 kCudnnBatchNormForwardInferenceCallTarget && in PropagateOperandConstraint()374 kCudnnBatchNormForwardInferenceCallTarget) { in PropagateBufferConstraint()
94 extern const char* const kCudnnBatchNormForwardInferenceCallTarget;
106 batch_norm_shape, operands, kCudnnBatchNormForwardInferenceCallTarget); in HandleBatchNormInference()
164 const char* const kCudnnBatchNormForwardInferenceCallTarget = variable176 return target == kCudnnBatchNormForwardInferenceCallTarget || in IsCustomCallToDnnBatchNorm()
151 kCudnnBatchNormForwardInferenceCallTarget)); in TEST_F()
881 if (target == xla::gpu::kCudnnBatchNormForwardInferenceCallTarget) { in EmitDnnBatchNorm()