Searched defs:inner_dim (Results 1 – 7 of 7) sorted by relevance
60 auto inner_dim = in0.dim_size(in0.dims() - 1); in Compute() local
938 int64 inner_dim = 1; in Compute() local1278 int64 inner_dim = 1; in Compute() local1429 int64 inner_dim = 1; in Compute() local1718 int64 inner_dim = 1; in Compute() local2070 int64 inner_dim = 1; in Compute() local
38 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); in Compile() local
871 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
1523 auto inner_dim = c->Dim(input, rank - 1); in __anon2aeb9a7a1202() local
2386 [&](int64 product, int64 inner_dim) { in EmitFastConcatenate()
1116 int64 inner_dim = RandomDim(); in TEST_F() local