Home
last modified time | relevance | path

Searched refs:InstructionMatchesPattern (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_scatter_expander.cc26 bool GpuScatterExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern() function in xla::GpuScatterExpander
Dgpu_scatter_expander.h33 bool InstructionMatchesPattern(HloInstruction* inst) override;
/external/tensorflow/tensorflow/compiler/xla/service/
Doperand_upcaster.h31 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dstable_sort_expander.h35 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dconvert_operand_folding.h33 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dconvolution_4d_expander.h31 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dop_expander_pass.h34 virtual bool InstructionMatchesPattern(HloInstruction* instruction) = 0;
Dcomparison_expander.h37 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Drng_expander.h28 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dgather_expander.h51 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dscatter_expander.h50 bool InstructionMatchesPattern(HloInstruction* inst) override;
Dlogistic_expander.h42 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dall_to_all_decomposer.h40 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dop_expander_pass.cc33 [&](HloInstruction* inst) { return InstructionMatchesPattern(inst); }); in Run()
Dcholesky_expander.h30 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dqr_expander.h30 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dconvert_operand_folding.cc34 bool ConvertOperandFolding::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::ConvertOperandFolding
Dtriangular_solve_expander.h34 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Drng_bit_generator_expander.h59 bool InstructionMatchesPattern(HloInstruction* instruction) override;
Dlogistic_expander.cc74 bool LogisticExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::LogisticExpander
Doperand_upcaster.cc44 bool OperandUpcaster::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern() function in xla::OperandUpcaster
Drng_bit_generator_expander.cc51 bool RngBitGeneratorExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::RngBitGeneratorExpander
Dcomparison_expander.cc63 bool ComparisonExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::ComparisonExpander
Dall_to_all_decomposer.cc37 bool AllToAllDecomposer::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::AllToAllDecomposer
Dconvolution_4d_expander.cc31 bool Convolution4DExpander::InstructionMatchesPattern( in InstructionMatchesPattern() function in xla::Convolution4DExpander

12