Home
last modified time | relevance | path

Searched defs:computation_map (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc285 const ComputationMap& computation_map, in SupportSpatialPartitioning()
578 const ComputationMap& computation_map, in InferShardingFromOperands()
1408 const ComputationMap& computation_map, in InferShardingFromUsers()
1667 ComputationMap computation_map; in Run() local
Dhlo_module.cc363 absl::flat_hash_map<int64, HloComputation*> computation_map; in CreateFromProto() local
Dhlo_computation.cc664 const absl::flat_hash_map<int64, HloComputation*>& computation_map, in CreateFromProto()
Dhlo_instruction.cc70 const absl::flat_hash_map<int64, HloComputation*>& computation_map, in CreateFromProto()