Searched defs:in_dim (Results 1 – 5 of 5) sorted by relevance
134 def _valid(in_dim, k_dim, s_dim): argument146 def _same(in_dim, s_dim): argument
1211 const int64 in_dim = out_dim + offset; in ComputeFlatInnerDims() local1214 for (int64 in_dim = 0; in_dim < offset; ++in_dim) { in ComputeFlatInnerDims() local1226 for (int64 in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims() local
759 size_t in_dim; member
1327 DSizes<ptrdiff_t, 4> in_dim; in TEST() local
490 auto in_dim = c->Dim(in, in_rank - i - 1); in __anon7c94107b0702() local