Searched defs:out_dim (Results 1 – 5 of 5) sorted by relevance
155 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local
1274 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local1287 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
170 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
1307 DSizes<ptrdiff_t, 3> out_dim; in TEST() local
500 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon42d741190702() local