Home
last modified time | relevance | path

Searched defs:InstructionMatchesPattern (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_scatter_expander.cc26 bool GpuScatterExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::GpuScatterExpander
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreal_imag_expander.cc22 bool RealImagExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::RealImagExpander
H A Dconvolution_pred_expander.cc32 bool ConvolutionPredExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::ConvolutionPredExpander
H A Dconvert_operand_folding.cc35 bool ConvertOperandFolding::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::ConvertOperandFolding
H A Dresult_caster.cc44 bool ResultCaster::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::ResultCaster
H A Doperand_upcaster.cc44 bool OperandUpcaster::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::OperandUpcaster
H A Dlogistic_expander.cc74 bool LogisticExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::LogisticExpander
H A Dbitcast_dtypes_expander.cc121 bool BitcastDtypesExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::BitcastDtypesExpander
H A Dall_to_all_decomposer.cc37 bool AllToAllDecomposer::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::AllToAllDecomposer
H A Drng_bit_generator_expander.cc51 bool RngBitGeneratorExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::RngBitGeneratorExpander
H A Dgather_simplifier.cc128 bool GatherSimplifier::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::GatherSimplifier
H A Dcomparison_expander.cc63 bool ComparisonExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::ComparisonExpander
H A Dstable_sort_expander.cc199 bool StableSortExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::StableSortExpander
H A Dconvolution_4d_expander.cc31 bool Convolution4DExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::Convolution4DExpander
H A Dscatter_simplifier.cc205 bool ScatterSimplifier::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::ScatterSimplifier
H A Drng_expander.cc128 bool RngExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::RngExpander
H A Dselect_and_scatter_expander.cc195 bool SelectAndScatterExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::SelectAndScatterExpander
H A Dcholesky_expander.cc216 bool CholeskyExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::CholeskyExpander
H A Dgather_expander.cc406 bool GatherExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::GatherExpander
H A Dscatter_expander.cc487 bool ScatterExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::ScatterExpander
H A Dqr_expander.cc505 bool QrExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::QrExpander
H A Dtriangular_solve_expander.cc550 bool TriangularSolveExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::TriangularSolveExpander
H A Deigh_expander.cc540 bool EighExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::EighExpander