Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h211 Status HandleConcatenate(HloInstruction* concatenate) override;
Dhlo_cost_analysis.h64 Status HandleConcatenate(const HloInstruction* concatenate) override;
Ddfs_hlo_visitor_with_default.h77 Status HandleConcatenate(HloInstructionPtr concatenate) override { in HandleConcatenate() function
Dhlo_verifier.h49 Status HandleConcatenate(HloInstruction* concatenate) override;
Ddfs_hlo_visitor.h86 virtual Status HandleConcatenate(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc121 Status HandleConcatenate(HloInstruction* hlo) override;
590 Status DynamicDimensionInferenceVisitor::HandleConcatenate( in HandleConcatenate() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc299 Status HloCostAnalysis::HandleConcatenate(const HloInstruction*) { in HandleConcatenate() function in xla::HloCostAnalysis
Dhlo_verifier.cc114 Status ShapeVerifier::HandleConcatenate(HloInstruction* concatenate) { in HandleConcatenate() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc293 Status HandleConcatenate(HloInstruction* concatenate) override;
1208 Status AlgebraicSimplifierVisitor::HandleConcatenate( in HandleConcatenate() function in xla::AlgebraicSimplifierVisitor
1310 return HandleConcatenate(replacement); in HandleConcatenate()
Dhlo_evaluator.cc487 Status HloEvaluator::HandleConcatenate(HloInstruction* concatenate) { in HandleConcatenate() function in xla::HloEvaluator
Dhlo_instruction.cc3107 return visitor->HandleConcatenate(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h165 Status HandleConcatenate(HloInstruction* concatenate) override;
Dir_emitter.cc2616 Status IrEmitter::HandleConcatenate(HloInstruction* concatenate) { in HandleConcatenate() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h432 Status HandleConcatenate(HloInstruction* hlo) override;
Dspmd_partitioner.cc1485 Status SpmdPartitioningVisitor::HandleConcatenate(HloInstruction* hlo) { in HandleConcatenate() function in xla::spmd::SpmdPartitioningVisitor