Home
last modified time | relevance | path

Searched refs:ExpandInstruction (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Doperand_upcaster.h33 StatusOr<HloInstruction*> ExpandInstruction(
Dstable_sort_expander.h36 StatusOr<HloInstruction*> ExpandInstruction(
Dconvert_operand_folding.h35 StatusOr<HloInstruction*> ExpandInstruction(
Dconvolution_4d_expander.h33 StatusOr<HloInstruction*> ExpandInstruction(
Dop_expander_pass.h39 virtual StatusOr<HloInstruction*> ExpandInstruction(
Dcomparison_expander.h41 StatusOr<HloInstruction*> ExpandInstruction(
Drng_expander.h30 StatusOr<HloInstruction*> ExpandInstruction(HloInstruction* rng) override;
Dgather_expander.h53 StatusOr<HloInstruction*> ExpandInstruction(
Dscatter_expander.h52 StatusOr<HloInstruction*> ExpandInstruction(HloInstruction* scatter) override;
Dlogistic_expander.h46 StatusOr<HloInstruction*> ExpandInstruction(
Dall_to_all_decomposer.h41 StatusOr<HloInstruction*> ExpandInstruction(
Dop_expander_pass.cc38 ExpandInstruction(inst)); in Run()
Dcholesky_expander.h32 StatusOr<HloInstruction*> ExpandInstruction(
Dqr_expander.h32 StatusOr<HloInstruction*> ExpandInstruction(
Dconvert_operand_folding.cc48 StatusOr<HloInstruction*> ConvertOperandFolding::ExpandInstruction( in ExpandInstruction() function in xla::ConvertOperandFolding
Dtriangular_solve_expander.h36 StatusOr<HloInstruction*> ExpandInstruction(
Drng_bit_generator_expander.h60 StatusOr<HloInstruction*> ExpandInstruction(HloInstruction* hlo) override;
Dlogistic_expander.cc78 StatusOr<HloInstruction*> LogisticExpander::ExpandInstruction( in ExpandInstruction() function in xla::LogisticExpander
Doperand_upcaster.cc57 StatusOr<HloInstruction*> OperandUpcaster::ExpandInstruction( in ExpandInstruction() function in xla::OperandUpcaster
Drng_bit_generator_expander.cc101 StatusOr<HloInstruction*> RngBitGeneratorExpander::ExpandInstruction( in ExpandInstruction() function in xla::RngBitGeneratorExpander
Dcomparison_expander.cc76 StatusOr<HloInstruction*> ComparisonExpander::ExpandInstruction( in ExpandInstruction() function in xla::ComparisonExpander
Dall_to_all_decomposer.cc55 StatusOr<HloInstruction*> AllToAllDecomposer::ExpandInstruction( in ExpandInstruction() function in xla::AllToAllDecomposer
Dconvolution_4d_expander.cc56 StatusOr<HloInstruction*> Convolution4DExpander::ExpandInstruction( in ExpandInstruction() function in xla::Convolution4DExpander
Dstable_sort_expander.cc37 StatusOr<HloInstruction*> StableSortExpander::ExpandInstruction( in ExpandInstruction() function in xla::StableSortExpander
Drng_expander.cc130 StatusOr<HloInstruction*> RngExpander::ExpandInstruction(HloInstruction* rng) { in ExpandInstruction() function in xla::RngExpander

12