Searched defs:dim_idx (Results 1 – 7 of 7) sorted by relevance
51 for (int dim_idx = 0; dim_idx < ndims_input; ++dim_idx) in ExpandInputDimsToOutputShape() local
96 int dim_idx = 0; in ReduceTransposeDimensions() local
136 for (int64_t dim_idx = 0; dim_idx < padded->shape().dimensions_size(); in RewriteReduction() local
400 Tensor dim_idx(DT_INT32, TensorShape({})); in SparsifyGatherInternal() local
1167 for (int64_t dim_idx = 0; dim_idx < params_rank; ++dim_idx) { in matchAndRewrite() local1177 for (int64_t dim_idx = 0; dim_idx < params_rank; ++dim_idx) { in matchAndRewrite() local1198 for (int64_t dim_idx = 0; dim_idx < params_rank; dim_idx++) { in matchAndRewrite() local4902 for (int64_t dim_idx = 0; dim_idx < input_rank; ++dim_idx) { in matchAndRewrite() local4914 for (int64_t dim_idx = 0; dim_idx < input_rank; ++dim_idx) { in matchAndRewrite() local5828 for (int64_t dim_idx = 0; dim_idx < value_rank; ++dim_idx) { in matchAndRewrite() local
1912 for (int64_t dim_idx = 0; dim_idx < input_rank; dim_idx++) { in matchAndRewrite() local
3046 for (int dim_idx = 0; dim_idx < sharded_rank; ++dim_idx) { in ComputeShardedArgShapes() local