Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc297 const ComputationMap& computation_map, in SupportSpatialPartitioning()
595 const ComputationMap& computation_map, in InferShardingFromOperands()
1427 const ComputationMap& computation_map, in InferShardingFromUsers()
1684 ComputationMap computation_map; in Run() local
Dhlo_module.cc370 absl::flat_hash_map<int64, HloComputation*> computation_map; in CreateFromProto() local
Dhlo_computation.cc685 const absl::flat_hash_map<int64, HloComputation*>& computation_map, in CreateFromProto()
Dhlo_instruction.cc71 const absl::flat_hash_map<int64, HloComputation*>& computation_map, in CreateFromProto()
/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc165 absl::flat_hash_map<int64, HloComputation*> computation_map; in Evaluate() local