Searched defs:out_dim (Results 1 – 6 of 6) sorted by relevance
157 DimensionHandle out_dim = c.Dim(out, j); in InferShapes() local
1210 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims() local1223 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims() local
39 auto* out_dim = out->add_dim(); in ProtoFromShapeHandle() local
166 int out_dim = dim + indices_in.dims() - 1; in MakeSplits() local
1307 DSizes<ptrdiff_t, 3> out_dim; in TEST() local
494 auto out_dim = c->Dim(out, out_rank - i - 1); in __anon7c94107b0702() local