Home
last modified time | relevance | path

Searched refs:HandleParameter (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.h53 Status HandleParameter(HloInstruction* parameter) override;
Dir_emitter_nested.cc112 Status IrEmitterNested::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::gpu::IrEmitterNested
Dir_emitter.h91 Status HandleParameter(HloInstruction* parameter) override;
Dir_emitter.cc660 Status IrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.h77 Status HandleParameter(HloInstruction* parameter) override;
Dfused_ir_emitter.cc144 Status FusedIrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::FusedIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_inliner.cc70 Status HandleParameter(HloInstruction* parameter) override { in HandleParameter() function in xla::__anona1e1e8bc0111::SubcomputationInsertionVisitor
Ddynamic_dimension_inference.cc42 Status HandleParameter(HloInstruction* hlo) override;
528 Status DynamicDimensionInferenceVisitor::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.h85 Status HandleParameter(const HloInstruction* parameter) override;
Ddfs_hlo_visitor_with_default.h136 Status HandleParameter(HloInstructionPtr parameter) override { in HandleParameter() function
Dhlo_evaluator.h191 Status HandleParameter(HloInstruction* parameter) override;
Dhlo_verifier.h74 Status HandleParameter(HloInstruction*) override;
Ddfs_hlo_visitor.h233 virtual Status HandleParameter(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc152 Status HloCostAnalysis::HandleParameter(const HloInstruction*) { in HandleParameter() function in xla::HloCostAnalysis
Dhlo_verifier.cc509 Status ShapeVerifier::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::ShapeVerifier
Dhlo_evaluator.cc417 Status HloEvaluator::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::HloEvaluator
Dhlo_instruction_test.cc58 Status HandleParameter(HloInstruction* parameter) override { in HandleParameter() function in xla::__anon0e998cf30111::OpAndUserCollectingVisitor
Dhlo_instruction.cc2498 return visitor->HandleParameter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc50 Status HandleParameter(const HloInstruction* parameter) override { in HandleParameter() function in xla::__anonc6eaaf4e0111::ExtractionVisitor
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h161 Status HandleParameter(HloInstruction* parameter) override;
Dcpu_compiler.cc223 Status HandleParameter(HloInstruction*) override { return Status::OK(); } in HandleParameter() function in xla::cpu::__anonaf93c4290211::CollectProfileCandidates
Dir_emitter.cc1350 Status IrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::cpu::IrEmitter