Searched defs:source_dim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.h | 186 int64 source_dim() const { return source_dim_; } in source_dim() function
194 explicit ScalarIndexedArray(Array* source, Array* indices, int64 source_dim, in ScalarIndexedArray()
225 int64 source_dim, in ScalarIndexedConstantArray()
337 Array* source, Array* indices, int64 source_dim, in ConstructScalarIndexedArray()
|
| D | indexed_array_analysis.cc | 188 ScalarIndexedArray* source, Array* indices, int64 source_dim, in FoldGatherOfGather()
290 int64 source_dim = dim_numbers.start_index_map(0); in ComputeArrayForGather() local
|
| D | sharding_propagation.cc | 756 const int64 source_dim = std::distance(dimensions.begin(), it); in InferShardingFromOperands() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1099 int64 source_dim = source_target_dims[0].first; in ReshardWithAllToAll() local
|