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
1162 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier2488 Status HandleCustomCall(HloInstruction* hlo) override { in HandleCustomCall() function in xla::__anon411d6f1b0e11::InstructionVerifier
511 Status TuplePointsToAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::TuplePointsToAnalysis
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::__anone66dd8310111::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