Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc116 Status FusedIrEmitter::HandleParameter(const HloInstruction* parameter) { in HandleParameter() function in xla::FusedIrEmitter
181 TF_RETURN_IF_ERROR(HandleParameter(instr)); in GetGenerator()
Dfused_ir_emitter.h85 Status HandleParameter(const HloInstruction* parameter);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.h52 Status HandleParameter(HloInstruction* parameter) override;
Dir_emitter.h90 Status HandleParameter(HloInstruction* parameter) override;
Dir_emitter_nested.cc162 Status IrEmitterNested::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::gpu::IrEmitterNested
Dir_emitter.cc572 Status IrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_inliner.cc69 Status HandleParameter(HloInstruction* parameter) override { in HandleParameter() function in xla::__anon11ac799a0111::SubcomputationInsertionVisitor
Dhlo_evaluator.h207 Status HandleParameter(HloInstruction* parameter) override;
Dhlo_cost_analysis.h94 Status HandleParameter(const HloInstruction* parameter) override;
Ddfs_hlo_visitor_with_default.h155 Status HandleParameter(HloInstructionPtr parameter) override { in HandleParameter() function
Dhlo_verifier.h83 Status HandleParameter(HloInstruction*) override;
Ddfs_hlo_visitor.h250 virtual Status HandleParameter(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc91 Status HandleParameter(HloInstruction* hlo) override;
1537 Status DynamicDimensionInferenceVisitor::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::DynamicDimensionInferenceVisitor
Ddynamic_padder.cc1573 Status HandleParameter(HloInstruction* hlo) override;
1755 Status DynamicShapeRemovingVisitor::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::__anoncf57f0820111::DynamicShapeRemovingVisitor
Dhlo_cost_analysis.cc208 Status HloCostAnalysis::HandleParameter(const HloInstruction*) { in HandleParameter() function in xla::HloCostAnalysis
Dhlo_verifier.cc744 Status ShapeVerifier::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::ShapeVerifier
Dhlo_instruction_test.cc60 Status HandleParameter(HloInstruction* parameter) override { in HandleParameter() function in xla::__anonabde1bd10111::OpAndUserCollectingVisitor
Dhlo_evaluator.cc452 Status HloEvaluator::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::HloEvaluator
Dhlo_instruction.cc3079 return visitor->HandleParameter(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc51 Status HandleParameter(const HloInstruction* parameter) override { in HandleParameter() function in xla::__anonf1179c6c0111::ExtractionVisitor
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.h185 tensorflow::Status HandleParameter(const xla::HloInstruction* instr) final { in HandleParameter() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h147 Status HandleParameter(HloInstruction* parameter) override;
Dcpu_compiler.cc259 Status HandleParameter(HloInstruction*) override { return Status::OK(); } in HandleParameter() function in xla::cpu::__anon31d794070311::CollectProfileCandidates
Dir_emitter.cc1315 Status IrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h422 Status HandleParameter(HloInstruction* hlo) override;

12