Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h149 Status HandleDynamicSlice(const HloInstruction* dynamic_slice) override;
Ddfs_hlo_visitor_with_default.h183 Status HandleDynamicSlice(HloInstructionPtr dynamic_slice) override { in HandleDynamicSlice() function
Dalgebraic_simplifier.h331 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
Ddfs_hlo_visitor.h261 virtual Status HandleDynamicSlice(HloInstructionPtr hlo) = 0;
Dhlo_verifier.h189 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
Ddynamic_dimension_inference.cc159 Status HandleDynamicSlice(HloInstruction* hlo) override;
1305 Status DynamicDimensionInferenceVisitor::HandleDynamicSlice( in HandleDynamicSlice() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc258 Status HloCostAnalysis::HandleDynamicSlice( in HandleDynamicSlice() function in xla::HloCostAnalysis
Dhlo_verifier.cc1215 Status ShapeVerifier::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() function in xla::ShapeVerifier
Dhlo_evaluator_typed_visitor.h1415 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override {
Dhlo_instruction.cc3642 return visitor->HandleDynamicSlice(this); in Visit()
Dalgebraic_simplifier.cc4805 Status AlgebraicSimplifierVisitor::HandleDynamicSlice( in HandleDynamicSlice() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h162 Status HandleDynamicSlice(HloInstruction* dynamic_slice) override;
Dir_emitter.cc2080 Status IrEmitter::HandleDynamicSlice(HloInstruction* dynamic_slice) { in HandleDynamicSlice() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h476 Status HandleDynamicSlice(HloInstruction* hlo) override;
Dspmd_partitioner.cc2987 Status SpmdPartitioningVisitor::HandleDynamicSlice(HloInstruction* hlo) { in HandleDynamicSlice() function in xla::spmd::SpmdPartitioningVisitor