Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc82 Status FusedIrEmitter::HandleConstant(const HloInstruction* constant) { in HandleConstant() function in xla::FusedIrEmitter
175 TF_RETURN_IF_ERROR(HandleConstant(instr)); in GetGenerator()
Dfused_ir_emitter.h81 Status HandleConstant(const HloInstruction* constant);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h79 Status HandleConstant(HloInstruction* constant) override;
Dir_emitter_unnested.h166 Status HandleConstant(HloInstruction* constant) override;
Dir_emitter.cc166 Status IrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc736 Status IrEmitterUnnested::HandleConstant(HloInstruction* constant) { in HandleConstant() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h209 Status HandleConstant(HloInstruction* constant) override;
Dhlo_cost_analysis.h55 Status HandleConstant(const HloInstruction* constant) override;
Ddfs_hlo_visitor_with_default.h146 Status HandleConstant(HloInstructionPtr constant) override { in HandleConstant() function
Dhlo_verifier.h75 Status HandleConstant(HloInstruction* constant) override;
Ddfs_hlo_visitor.h241 virtual Status HandleConstant(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc216 Status HloCostAnalysis::HandleConstant(const HloInstruction*) { in HandleConstant() function in xla::HloCostAnalysis
Dhlo_verifier.cc596 Status ShapeVerifier::HandleConstant(HloInstruction* constant) { in HandleConstant() function in xla::ShapeVerifier
Dhlo_instruction_test.cc66 Status HandleConstant(HloInstruction* constant) override { in HandleConstant() function in xla::__anonabde1bd10111::OpAndUserCollectingVisitor
Dhlo_evaluator.cc471 Status HloEvaluator::HandleConstant(HloInstruction*) { return Status::OK(); } in HandleConstant() function in xla::HloEvaluator
Dalgebraic_simplifier.cc295 Status HandleConstant(HloInstruction* constant) override;
1388 Status AlgebraicSimplifierVisitor::HandleConstant(HloInstruction* constant) { in HandleConstant() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3075 return visitor->HandleConstant(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h134 Status HandleConstant(HloInstruction* constant) override;
Dcpu_compiler.cc257 Status HandleConstant(HloInstruction*) override { return Status::OK(); } in HandleConstant() function in xla::cpu::__anon31d794070311::CollectProfileCandidates
Dir_emitter.cc274 Status IrEmitter::HandleConstant(HloInstruction* constant) { in HandleConstant() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h411 Status HandleConstant(HloInstruction* hlo) override;
Dspmd_partitioner.cc2264 Status SpmdPartitioningVisitor::HandleConstant(HloInstruction* hlo) { in HandleConstant() function in xla::spmd::SpmdPartitioningVisitor