Home
last modified time | relevance | path

Searched refs:HandleCustomCall (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlogical_buffer_analysis.h71 Status HandleCustomCall(HloInstruction* custom_call) override;
Dlogical_buffer_analysis.cc192 Status LogicalBufferAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::LogicalBufferAnalysis
Dtuple_points_to_analysis.h259 Status HandleCustomCall(HloInstruction* custom_call) override;
Dhlo_evaluator.h265 Status HandleCustomCall(HloInstruction* custom_call) override;
Dhlo_cost_analysis.h103 Status HandleCustomCall(const HloInstruction* custom_call) override;
Ddfs_hlo_visitor_with_default.h164 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
Dhlo_verifier.h86 Status HandleCustomCall(HloInstruction*) override;
Ddfs_hlo_visitor.h253 virtual Status HandleCustomCall(HloInstructionPtr hlo) = 0;
Dtuple_points_to_analysis.cc478 Status TuplePointsToAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::TuplePointsToAnalysis
Ddynamic_dimension_inference.cc109 Status HandleCustomCall(HloInstruction* hlo) override;
245 Status DynamicDimensionInferenceVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::DynamicDimensionInferenceVisitor
Ddynamic_padder.cc1568 Status HandleCustomCall(HloInstruction* hlo) override;
1759 Status DynamicShapeRemovingVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::__anoncf57f0820111::DynamicShapeRemovingVisitor
Dhlo_cost_analysis.cc936 Status HloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloCostAnalysis
Dhlo_verifier.cc792 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier
Dhlo_evaluator.cc2515 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
Dhlo_instruction.cc3239 return visitor->HandleCustomCall(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h96 Status HandleCustomCall(HloInstruction* custom_call) override;
Dir_emitter_unnested.h174 Status HandleCustomCall(HloInstruction* custom_call) override;
Dir_emitter.cc598 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1062 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h163 Status HandleCustomCall(HloInstruction* custom_call) override;
Dir_emitter.cc2295 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h412 Status HandleCustomCall(HloInstruction* hlo) override;
Dspmd_partitioner.cc1775 Status SpmdPartitioningVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::spmd::SpmdPartitioningVisitor