Searched refs:kCudnnBatchNormForwardTrainingCallTarget (Results 1 – 7 of 7) sorted by relevance
83 extern const char* const kCudnnBatchNormForwardTrainingCallTarget;
111 const char* const kCudnnBatchNormForwardTrainingCallTarget = variable122 target == kCudnnBatchNormForwardTrainingCallTarget || in IsCustomCallToDnnBatchNorm()
295 kCudnnBatchNormForwardTrainingCallTarget && in PropagateOperandConstraint()348 kCudnnBatchNormForwardTrainingCallTarget && in PropagateBufferConstraint()
121 kCudnnBatchNormForwardTrainingCallTarget)); in HandleBatchNormTraining()
154 CHECK_EQ(hlo->custom_call_target(), kCudnnBatchNormForwardTrainingCallTarget); in CudnnBatchNormForwardTrainingThunk()
218 kCudnnBatchNormForwardTrainingCallTarget)); in TEST_F()
405 kCudnnBatchNormForwardTrainingCallTarget) { in HandleCustomCall()