Searched defs:HandleCustomCall (Results 1 – 12 of 12) sorted by relevance
39 Status HandleCustomCall(HloInstruction *instr) override { in HandleCustomCall() function in xla::gpu::GemmBroadcastFoldingVisitor
593 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
192 Status LogicalBufferAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::LogicalBufferAnalysis
179 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
478 Status TuplePointsToAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::TuplePointsToAnalysis
965 Status HloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloCostAnalysis
1129 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier
1960 Status DynamicShapeRemovingVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::__anonab86f1e70111::DynamicShapeRemovingVisitor
247 Status DynamicDimensionInferenceVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::DynamicDimensionInferenceVisitor
2579 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
374 Status SpmdPartitioningVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::spmd::SpmdPartitioningVisitor
2301 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter