Searched defs:HloComputation (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 85 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
190 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
227 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
303 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
345 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
390 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
508 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
569 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
633 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
674 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
[all …]
|
| D | hlo_memory_scheduler.cc | 81 const absl::flat_hash_map<const HloComputation*, int64>& in Run()
106 const absl::flat_hash_map<const HloComputation*, int64>& in ListScheduler()
406 const absl::flat_hash_map<const HloComputation*, int64>& in ScheduleComputationHelper()
423 const absl::flat_hash_map<const HloComputation*, int64>& in DFSMemoryScheduler()
494 const absl::flat_hash_map<const HloComputation*, int64>& in ListMemoryScheduler()
504 const absl::flat_hash_map<const HloComputation*, int64>& in PostOrderMemoryScheduler()
513 const absl::flat_hash_map<const HloComputation*, int64>& in DefaultMemoryScheduler()
|
| D | hlo_execution_profile.cc | 79 const std::pair<const HloComputation*, int64>& right) { in CreateHloProfilePrinterData()
|
| D | hlo_clone_context.h | 27 class HloComputation; variable
|
| D | heap_simulator.cc | 61 const absl::flat_hash_map<const HloComputation*, int64>* in MinimumMemoryForComputation()
93 const absl::flat_hash_map<const HloComputation*, int64>* in Run()
353 const absl::flat_hash_map<const HloComputation*, int64>* in HeapSimulator()
530 const absl::flat_hash_map<const HloComputation*, int64>& in AccountForSubcomputationMemory()
|
| D | hlo_instruction.h | 63 class HloComputation; variable
1704 friend class HloComputation; variable
|
| D | heap_simulator.h | 234 const absl::flat_hash_map<const HloComputation*, int64>& in AccountForSubcomputationMemory()
|
| D | dfs_hlo_visitor.h | 36 class HloComputation; variable
|
| D | dfs_hlo_visitor_with_default.h | 32 class HloComputation; variable
|
| D | hlo_module.cc | 126 const std::unordered_map<HloComputation*, HloComputation*>& replacements) { in ReplaceComputations()
|
| D | buffer_assignment.cc | 900 flat_hash_map<const HloComputation*, flat_hash_set<const LogicalBuffer*>>* in AssignBuffersForComputation()
1147 const flat_hash_map<const HloComputation*, in AssignBuffersWithSequentialOrdering()
|
| D | hlo_computation.cc | 67 HloComputation::HloComputation( in HloComputation() function in xla::HloComputation
|
| D | hlo_instruction.cc | 1722 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 491 std::unordered_map<const HloComputation*, int64>* in CreateHloProfilingArtifacts()
|
| D | ir_emitter.cc | 87 std::unordered_map<const HloComputation*, int64> computation_to_profile_idx, in IrEmitter()
|