Searched defs:GetSliceIndices (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops.cc | 178 Eigen::DSizes<Eigen::DenseIndex, Rank> GetSliceIndices( in GetSliceIndices() function
186 Eigen::DSizes<Eigen::DenseIndex, 1> GetSliceIndices( in GetSliceIndices() function
195 Eigen::DSizes<Eigen::DenseIndex, 2> GetSliceIndices( in GetSliceIndices() function
205 Eigen::DSizes<Eigen::DenseIndex, 3> GetSliceIndices( in GetSliceIndices() function
218 Eigen::DSizes<Eigen::DenseIndex, 4> GetSliceIndices( in GetSliceIndices() function
235 Eigen::DSizes<Eigen::DenseIndex, 5> GetSliceIndices( in GetSliceIndices() function
256 Eigen::DSizes<Eigen::DenseIndex, 6> GetSliceIndices( in GetSliceIndices() function
282 Eigen::DSizes<Eigen::DenseIndex, 7> GetSliceIndices( in GetSliceIndices() function
313 Eigen::DSizes<Eigen::DenseIndex, 8> GetSliceIndices( in GetSliceIndices() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sharding_util_ops.cc | 97 std::vector<int64_t> GetSliceIndices(absl::Span<const int64> num_partitions, in GetSliceIndices() function
|