Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.h61 Status HandleParameter(HloInstruction* parameter) override;
Dir_emitter.h86 Status HandleParameter(HloInstruction* parameter) override;
Dir_emitter_nested.cc175 Status IrEmitterNested::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::gpu::IrEmitterNested
Dir_emitter.cc554 Status IrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc52 Status HandleParameter(const HloInstruction* parameter) override { in HandleParameter() function in xla::__anon1b44898a0111::ExtractionVisitor
/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_inliner.cc72 Status HandleParameter(HloInstruction* parameter) override { in HandleParameter() function in xla::__anon81770a780111::SubcomputationInsertionVisitor
Dhlo_cost_analysis.h138 Status HandleParameter(const HloInstruction* parameter) override;
Ddfs_hlo_visitor_with_default.h168 Status HandleParameter(HloInstructionPtr parameter) override { in HandleParameter() function
Dhlo_evaluator.h275 Status HandleParameter(HloInstruction* parameter) override;
Ddfs_hlo_visitor.h256 virtual Status HandleParameter(HloInstructionPtr hlo) = 0;
Dhlo_verifier.h184 Status HandleParameter(HloInstruction*) override;
Ddynamic_dimension_inference.cc105 Status HandleParameter(HloInstruction* hlo) override;
1813 Status DynamicDimensionInferenceVisitor::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::DynamicDimensionInferenceVisitor
Ddynamic_padder.cc1913 Status HandleParameter(HloInstruction* hlo) override;
2103 Status DynamicShapeRemovingVisitor::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::__anon5c0a30e00111::DynamicShapeRemovingVisitor
Dhlo_cost_analysis.cc213 Status HloCostAnalysis::HandleParameter(const HloInstruction*) { in HandleParameter() function in xla::HloCostAnalysis
Dhlo_instruction_test.cc61 Status HandleParameter(HloInstruction* parameter) override { in HandleParameter() function in xla::__anon4922aaaf0111::OpAndUserCollectingVisitor
Dhlo_verifier.cc1114 Status ShapeVerifier::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::ShapeVerifier
Dhlo_evaluator.cc1085 Status HloEvaluator::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::HloEvaluator
Dhlo_instruction.cc3498 return visitor->HandleParameter(this); in Visit()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.h183 tensorflow::Status HandleParameter(const xla::HloInstruction* instr) final { in HandleParameter() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h155 Status HandleParameter(HloInstruction* parameter) override;
Dcpu_compiler.cc393 Status HandleParameter(HloInstruction*) override { return OkStatus(); } in HandleParameter() function in xla::cpu::__anonb41b63e50511::CollectProfileCandidates
Dir_emitter.cc1408 Status IrEmitter::HandleParameter(HloInstruction* parameter) { in HandleParameter() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h485 Status HandleParameter(HloInstruction* hlo) override;
Dspmd_partitioner.cc3456 Status SpmdPartitioningVisitor::HandleParameter(HloInstruction* hlo) { in HandleParameter() function in xla::spmd::SpmdPartitioningVisitor