Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc298 struct ConvWithConvertOrClamp { struct
307 absl::optional<ConvWithConvertOrClamp> FindConvWithClampAndConvertToInt8( in FindConvWithClampAndConvertToInt8()
330 return ConvWithConvertOrClamp{instr, gte, conv}; in FindConvWithClampAndConvertToInt8()
339 Status RewriteForConvertOrClampImpl(ConvWithConvertOrClamp match) { in RewriteForConvertOrClampImpl()
358 Status RewriteForFinalOutput(ConvWithConvertOrClamp match) { in RewriteForFinalOutput()
395 std::vector<ConvWithConvertOrClamp> matches; in RunFuseClamp()
402 for (const ConvWithConvertOrClamp& match : matches) { in RunFuseClamp()
431 absl::optional<ConvWithConvertOrClamp> FindConvWithConvertToFloat( in FindConvWithConvertToFloat()
451 return ConvWithConvertOrClamp{instr, gte, conv}; in FindConvWithConvertToFloat()
463 std::vector<ConvWithConvertOrClamp> matches; in RunFuseConvertToFloat()
[all …]