Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmove_compilation_to_host.cc212 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp>& computation_map, in HandleCompilationOps()
382 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp> computation_map; in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dsharding_propagation.cc323 const ShardingPropagation::ComputationMap& computation_map, bool is_spmd, in SupportSpatialPartitioning()
719 const ShardingPropagation::ComputationMap& computation_map, in InferShardingFromUsers()
1707 HloInstruction* instruction, const ComputationMap& computation_map, in InferShardingFromOperands()
2321 ComputationMap computation_map; in Run() local
H A Dhlo_module.cc422 absl::flat_hash_map<int64_t, HloComputation*> computation_map; in CreateFromProto() local
H A Dhlo_computation.cc657 const absl::flat_hash_map<int64_t, HloComputation*>& computation_map, in CreateFromProto()
H A Dhlo_instruction.cc87 const absl::flat_hash_map<int64_t, HloComputation*>& computation_map, in CreateFromProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dvalue_inference.cc171 absl::flat_hash_map<int64_t, HloComputation*> computation_map; in Evaluate() local