Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h104 Status HandleSlice(const HloInstruction* slice) override;
Ddfs_hlo_visitor_with_default.h167 Status HandleSlice(HloInstructionPtr slice) override { in HandleSlice() function
Dhlo_verifier.h87 Status HandleSlice(HloInstruction* slice) override;
Ddfs_hlo_visitor.h254 virtual Status HandleSlice(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc141 Status HandleSlice(HloInstruction* hlo) override;
1098 Status DynamicDimensionInferenceVisitor::HandleSlice(HloInstruction* hlo) { in HandleSlice() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc252 Status HloCostAnalysis::HandleSlice(const HloInstruction* slice) { in HandleSlice() function in xla::HloCostAnalysis
Dhlo_verifier.cc838 Status ShapeVerifier::HandleSlice(HloInstruction* slice) { in HandleSlice() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc351 Status HandleSlice(HloInstruction* slice) override;
4138 Status AlgebraicSimplifierVisitor::HandleSlice(HloInstruction* slice) { in HandleSlice() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator_typed_visitor.h2435 Status HandleSlice(HloInstruction* slice) override {
Dhlo_instruction.cc3213 return visitor->HandleSlice(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h153 Status HandleSlice(HloInstruction* slice) override;
Dir_emitter.cc1832 Status IrEmitter::HandleSlice(HloInstruction* slice) { in HandleSlice() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h434 Status HandleSlice(HloInstruction* hlo) override;
Dspmd_partitioner.cc1579 Status SpmdPartitioningVisitor::HandleSlice(HloInstruction* hlo) { in HandleSlice() function in xla::spmd::SpmdPartitioningVisitor