Searched defs:in_dim (Results 1 – 3 of 3) sorted by relevance
1275 const int64 in_dim = out_dim + offset; in ComputeFlatInnerDims() local1278 for (int64 in_dim = 0; in_dim < offset; ++in_dim) { in ComputeFlatInnerDims() local1290 for (int64 in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims() local
496 auto in_dim = c->Dim(in, in_rank - i - 1); in __anon42d741190702() local
1327 DSizes<ptrdiff_t, 4> in_dim; in TEST() local