Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc363 const auto is_one_to_one_X_to_Y_cast = [](const HloInstruction* instr, in RewriteForFinalOutput() local
376 } else if (is_one_to_one_X_to_Y_cast(match.conv->operand(3), S8, F32)) { in RewriteForFinalOutput()