| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instructions.cc | 123 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
228 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
325 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
397 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
447 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
524 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
566 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
615 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
649 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
772 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
[all …]
|
| H A D | hlo_memory_scheduler.cc | 84 const absl::flat_hash_map<const HloComputation*, int64_t>& in Run()
109 const absl::flat_hash_map<const HloComputation*, int64_t>& in ListScheduler()
419 const absl::flat_hash_map<const HloComputation*, int64_t>& in ScheduleComputationHelper()
441 const absl::flat_hash_map<const HloComputation*, int64_t>& in DFSMemoryScheduler()
563 const absl::flat_hash_map<const HloComputation*, int64_t>& in ListMemoryScheduler()
586 const absl::flat_hash_map<const HloComputation*, int64_t>& in PostOrderMemoryScheduler()
607 const absl::flat_hash_map<const HloComputation*, int64_t>& in DefaultMemoryScheduler()
|
| H A D | heap_simulator.cc | 76 const absl::flat_hash_map<const HloComputation*, int64_t>* in MinimumMemoryForComputation()
124 const absl::flat_hash_map<const HloComputation*, int64_t>* in Run()
341 const absl::flat_hash_map<const HloComputation*, int64_t>* in HeapSimulator()
467 const absl::flat_hash_map<const HloComputation*, int64_t>& in AccountForSubcomputationMemory()
|
| H A D | hlo_execution_profile.cc | 80 const std::pair<const HloComputation*, int64_t>& right) { in CreateHloProfilePrinterData()
|
| H A D | flatten_call_graph.cc | 68 [&](HloComputation*) { return new_computation; }); in ReplaceCalledComputation() argument
|
| H A D | hlo_clone_context.h | 27 class HloComputation; variable
|
| H A D | async_op_canonicalizer.cc | 113 [&](HloComputation*) { return computation; }); in Run() argument
|
| H A D | buffer_assignment.cc | 1156 absl::flat_hash_map<const HloComputation*, in AssignSingleHloBuffer()
1296 absl::flat_hash_map<const HloComputation*, in AssignBuffersForComputations()
1464 const flat_hash_map<const HloComputation*, flat_hash_set<const HloValue*>>& in AssignBuffersWithSequentialOrdering()
|
| H A D | hlo_dce.cc | 101 absl::flat_hash_map<HloComputation*, int>& live_call_counts) { in RecursivelyRemoveDeadComputation()
|
| H A D | heap_simulator.h | 267 const absl::flat_hash_map<const HloComputation*, int64_t>& in AccountForSubcomputationMemory()
|
| H A D | dfs_hlo_visitor.h | 34 class HloComputation; variable
|
| H A D | hlo_module.cc | 167 const absl::flat_hash_map<HloComputation*, HloComputation*>& replacements) { in ReplaceComputations()
|
| H A D | hlo_instruction.h | 65 class HloComputation; variable
|
| H A D | hlo_computation.cc | 75 HloComputation::HloComputation( in HloComputation() function in xla::HloComputation
|
| H A D | hlo_instructions.h | 407 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPathIgnoringChannelIdValues()
|
| H A D | hlo_instruction.cc | 2354 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalInternal()
2467 const std::function<bool(const HloComputation*, const HloComputation*)>& in IdenticalSlowPath()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.h | 40 class HloComputation; variable
136 std::unordered_map<const xla::HloComputation*, in HloFunctionImporter()
|
| H A D | hlo_module_importer.h | 33 class HloComputation; variable
|
| H A D | hlo_function_importer.cc | 244 std::unordered_map<const HloComputation*, FuncOp>* function_map, in ImportAsFunc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_compiler.cc | 786 absl::flat_hash_map<const HloComputation*, int64_t>* in CreateHloProfilingArtifacts()
|
| H A D | ir_emitter.cc | 101 absl::flat_hash_map<const HloComputation*, int64_t> in IrEmitter()
|