Home
last modified time | relevance | path

Searched defs:HandleCustomCall (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dlhlo_dialect_emitter.cc291 Status LhloDialectEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::mlir_gpu::LhloDialectEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk_emitter.cc133 Status ThunkEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::ThunkEmitter
Dir_emitter.cc832 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc368 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h152 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
Ddynamic_padder.cc958 Status DynamicShapeRemovingVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::__anon8bfa6aa60111::DynamicShapeRemovingVisitor
Dhlo_cost_analysis.cc854 Status HloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloCostAnalysis
Ddynamic_dimension_inference.cc180 Status DynamicDimensionInferenceVisitor::HandleCustomCall(HloInstruction* hlo) { in HandleCustomCall() function in xla::DynamicDimensionInferenceVisitor
Dhlo_verifier.cc683 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier
Dhlo_evaluator.cc2478 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2472 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter