| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 152 Status HloCostAnalysis::HandleParameter(const HloInstruction*) { in HandleParameter()
159 Status HloCostAnalysis::HandleConstant(const HloInstruction*) { in HandleConstant()
166 Status HloCostAnalysis::HandleIota(const HloInstruction*) { in HandleIota()
170 Status HloCostAnalysis::HandleGetTupleElement(const HloInstruction*) { in HandleGetTupleElement()
183 Status HloCostAnalysis::HandleTupleSelect(const HloInstruction*) { in HandleTupleSelect()
187 Status HloCostAnalysis::HandleReverse(const HloInstruction*) { in HandleReverse()
219 Status HloCostAnalysis::HandleConcatenate(const HloInstruction*) { in HandleConcatenate()
227 Status HloCostAnalysis::HandleCopy(const HloInstruction*) { in HandleCopy()
255 Status HloCostAnalysis::HandleInfeed(const HloInstruction*) { in HandleInfeed()
259 Status HloCostAnalysis::HandleOutfeed(const HloInstruction*) { in HandleOutfeed()
[all …]
|
| D | logical_buffer_analysis.cc | 111 Status LogicalBufferAnalysis::HandleGetTupleElement(HloInstruction*) { in HandleGetTupleElement() argument
130 Status LogicalBufferAnalysis::HandleBitcast(HloInstruction*) { in HandleBitcast() argument
136 Status LogicalBufferAnalysis::HandleDomain(HloInstruction*) { in HandleDomain() argument
|
| D | hlo_computation.cc | 319 absl::flat_hash_map<HloInstruction*, VisitState>* visited) const { in ComputeInstructionPostOrder()
741 [](const HloInstruction*, const HloInstruction*) { return true; }, in operator ==()
893 std::pair<const HloInstruction*, std::unique_ptr<HloInstruction>> r1, in CloneWithReplacementPairs()
903 std::pair<const HloInstruction*, std::unique_ptr<HloInstruction>> r1, in CloneWithReplacementPairs()
915 std::pair<const HloInstruction*, std::unique_ptr<HloInstruction>> r1, in CloneWithReplacementPairs()
929 absl::flat_hash_map<const HloInstruction*, std::unique_ptr<HloInstruction>> in CloneWithReplacements()
|
| D | transpose_folding.h | 38 static OperandIndices NeverFoldTranspose(const HloInstruction&, in NeverFoldTranspose()
|
| D | hlo_domain_metadata.h | 32 class HloInstruction; variable
|
| D | hlo_casting_utils.h | 26 class HloInstruction; variable
|
| D | map_inliner.cc | 42 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() argument
|
| D | hlo_clone_context.h | 26 class HloInstruction; variable
|
| D | hlo_execution_profile.h | 31 class HloInstruction; variable
|
| D | instruction_fusion.cc | 198 absl::flat_hash_map<std::pair<HloInstruction*, HloInstruction*>, bool>* in CanFuseOnAllPaths() argument
|
| D | dynamic_dimension_inference.cc | 126 int64 operand_index, HloInstruction* dynamic_size) { in HandleTuple() argument
293 HloInstruction*) { in HandleGetDimensionSize() argument
470 HloInstruction* dynamic_size) { in HandleWhile() argument
507 int64 operand_index, HloInstruction* dynamic_size) { in HandleWhile() argument
|
| D | hlo_verifier.h | 103 Status FinishVisit(HloInstruction*) override { return Status::OK(); } in FinishVisit() argument
|
| D | while_loop_invariant_code_motion.cc | 39 flat_hash_map<HloInstruction*, HloInstruction*>* hoisted_instructions, in CreateLoopInvariantCopy()
|
| D | hlo_evaluator.cc | 214 absl::make_unique<FunctionVisitor>([](HloInstruction*) { in HloEvaluator() argument
219 absl::make_unique<FunctionVisitor>([](HloInstruction*) { in HloEvaluator() argument
224 absl::make_unique<FunctionVisitor>([](HloInstruction*) { in HloEvaluator() argument
310 const std::unordered_map<const HloInstruction*, const Literal*>& in EvaluateWithSubstitutions()
434 Status HloEvaluator::HandleConstant(HloInstruction*) { return Status::OK(); } in HandleConstant() argument
|
| D | while_util.cc | 174 static StatusOr<std::unique_ptr<HloComputation>> MakeCountedLoopBodyComputation( in MakeCountedLoopBodyComputation()
|
| D | dfs_hlo_visitor.h | 37 class HloInstruction; variable
|
| D | dfs_hlo_visitor_with_default.h | 33 class HloInstruction; variable
|
| D | hlo_rematerialization.cc | 89 absl::flat_hash_map<const HloInstruction*, bool>* remat_able) { in CanBeRematerialized()
865 absl::flat_hash_map<const HloInstruction*, bool>* remat_able) { in PickRematerializationCandidate()
|
| D | convolution_group_converter.cc | 47 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() argument
|
| D | batchnorm_expander.cc | 52 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 122 Status IrEmitter::HandleSend(HloInstruction*) { in HandleSend() argument
126 Status IrEmitter::HandleSendDone(HloInstruction*) { in HandleSendDone() argument
130 Status IrEmitter::HandleRecv(HloInstruction*) { in HandleRecv() argument
134 Status IrEmitter::HandleRecvDone(HloInstruction*) { in HandleRecvDone() argument
138 Status IrEmitter::HandleScatter(HloInstruction*) { in HandleScatter() argument
744 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() argument
748 Status IrEmitter::HandleInfeed(HloInstruction*) { in HandleInfeed() argument
753 Status IrEmitter::HandleOutfeed(HloInstruction*) { in HandleOutfeed() argument
758 Status IrEmitter::HandleBatchNormInference(HloInstruction*) { in HandleBatchNormInference() argument
766 Status IrEmitter::HandleBatchNormTraining(HloInstruction*) { in HandleBatchNormTraining() argument
[all …]
|
| D | cudnn_batchnorm_rewriter.cc | 35 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() argument
|
| D | thunk_schedule.cc | 31 const absl::flat_hash_map<const HloInstruction*, Thunk*>& hlo_to_thunk) { in AddDependenciesOnTransitiveOperands()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 190 const std::unordered_map<const HloInstruction*, int64>& in GetCandidatesForComputation()
201 std::unordered_map<const HloInstruction*, int64>* hlo_to_profile_idx, in CollectProfileCandidates()
202 const std::unordered_map<const HloInstruction*, int64>& assigned_indices) in CollectProfileCandidates()
221 Status HandleConstant(HloInstruction*) override { return Status::OK(); } in HandleConstant() argument
223 Status HandleParameter(HloInstruction*) override { return Status::OK(); } in HandleParameter() argument
489 std::unordered_map<const HloInstruction*, int64>* in CreateHloProfilingArtifacts()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 80 Status FinishVisit(const HloInstruction* /*root*/) override { in FinishVisit()
|