Searched refs:kCudnnBatchNormForwardInferenceCallTarget (Results 1 – 7 of 7) sorted by relevance
82 extern const char* const kCudnnBatchNormForwardInferenceCallTarget;
109 const char* const kCudnnBatchNormForwardInferenceCallTarget = variable121 return target == kCudnnBatchNormForwardInferenceCallTarget || in IsCustomCallToDnnBatchNorm()
283 kCudnnBatchNormForwardInferenceCallTarget && in PropagateOperandConstraint()341 kCudnnBatchNormForwardInferenceCallTarget) { in PropagateBufferConstraint()
83 batch_norm->shape(), operands, kCudnnBatchNormForwardInferenceCallTarget); in HandleBatchNormInference()
95 kCudnnBatchNormForwardInferenceCallTarget); in CudnnBatchNormForwardInferenceThunk()
149 kCudnnBatchNormForwardInferenceCallTarget)); in TEST_F()
381 kCudnnBatchNormForwardInferenceCallTarget) { in HandleCustomCall()