Searched defs:HandleCustomCall (Results 1 – 13 of 13) sorted by relevance
24 Status GpuHloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::GpuHloCostAnalysis
40 Status HandleCustomCall(HloInstruction *instr) override { in HandleCustomCall() function in xla::gpu::GemmBroadcastFoldingVisitor
580 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
181 Status LogicalBufferAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::LogicalBufferAnalysis
177 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
511 Status TuplePointsToAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::TuplePointsToAnalysis
1162 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier2488 Status HandleCustomCall(HloInstruction* hlo) override { in HandleCustomCall() function in xla::__anonb51dd00a0e11::InstructionVerifier
994 Status HloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloCostAnalysis
2107 Status DynamicShapeRemovingVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::__anon5c0a30e00111::DynamicShapeRemovingVisitor
276 Status DynamicDimensionInferenceVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::DynamicDimensionInferenceVisitor
3821 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
376 Status SpmdPartitioningVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::spmd::SpmdPartitioningVisitor
2426 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter