Searched defs:HandleCustomCall (Results 1 – 12 of 12) sorted by relevance
192 Status LogicalBufferAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::LogicalBufferAnalysis
164 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
478 Status TuplePointsToAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::TuplePointsToAnalysis
936 Status HloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloCostAnalysis
792 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier
1759 Status DynamicShapeRemovingVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::__anoncf57f0820111::DynamicShapeRemovingVisitor
245 Status DynamicDimensionInferenceVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::DynamicDimensionInferenceVisitor
2515 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
598 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
1062 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::IrEmitterUnnested
2295 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter
1775 Status SpmdPartitioningVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::spmd::SpmdPartitioningVisitor