| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.h | 39 class HloComputation; variable
81 std::unordered_map<const xla::HloComputation*, in HloFunctionImporter()
|
| D | hlo_module_importer.h | 32 class HloComputation; variable
|
| D | hlo_function_importer.cc | 110 std::unordered_map<const HloComputation*, FuncOp>* function_map, in ImportAsFunc()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 88 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
193 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
233 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
283 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
359 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
401 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
450 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
484 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
605 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
656 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
[all …]
|
| D | hlo_memory_scheduler.cc | 84 const absl::flat_hash_map<const HloComputation*, int64>& in Run()
109 const absl::flat_hash_map<const HloComputation*, int64>& in ListScheduler()
419 const absl::flat_hash_map<const HloComputation*, int64>& in ScheduleComputationHelper()
440 const absl::flat_hash_map<const HloComputation*, int64>& in DFSMemoryScheduler()
547 const absl::flat_hash_map<const HloComputation*, int64>& in ListMemoryScheduler()
567 const absl::flat_hash_map<const HloComputation*, int64>& in PostOrderMemoryScheduler()
585 const absl::flat_hash_map<const HloComputation*, int64>& in DefaultMemoryScheduler()
|
| D | hlo_execution_profile.cc | 80 const std::pair<const HloComputation*, int64>& right) { in CreateHloProfilePrinterData()
|
| D | hlo_clone_context.h | 27 class HloComputation; variable
|
| D | heap_simulator.cc | 71 const absl::flat_hash_map<const HloComputation*, int64>* in MinimumMemoryForComputation()
121 const absl::flat_hash_map<const HloComputation*, int64>* in Run()
338 const absl::flat_hash_map<const HloComputation*, int64>* in HeapSimulator()
464 const absl::flat_hash_map<const HloComputation*, int64>& in AccountForSubcomputationMemory()
|
| D | heap_simulator.h | 261 const absl::flat_hash_map<const HloComputation*, int64>& in AccountForSubcomputationMemory()
|
| D | dfs_hlo_visitor.h | 36 class HloComputation; variable
|
| D | buffer_assignment.cc | 1100 absl::flat_hash_map<const HloComputation*, in AssignSingleHloBuffer()
1240 absl::flat_hash_map<const HloComputation*, in AssignBuffersForComputations()
1408 const flat_hash_map<const HloComputation*, flat_hash_set<const HloValue*>>& in AssignBuffersWithSequentialOrdering()
|
| D | hlo_module.cc | 159 const std::unordered_map<HloComputation*, HloComputation*>& replacements) { in ReplaceComputations()
|
| D | hlo_instruction.h | 63 class HloComputation; variable
|
| D | hlo_instructions.h | 252 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
|
| D | hlo_computation.cc | 67 HloComputation::HloComputation( in HloComputation() function in xla::HloComputation
|
| D | hlo_instruction.cc | 2015 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalInternal()
2114 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 541 std::unordered_map<const HloComputation*, int64>* in CreateHloProfilingArtifacts()
|
| D | ir_emitter.cc | 95 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx, in IrEmitter()
|