Searched defs:inner_dim (Results 1 – 9 of 9) sorted by relevance
60 auto inner_dim = in0.dim_size(in0.dims() - 1); in Compute() local
43 int64 inner_dim = input_dims[num_dims - 1]; in operator() local
1019 int64 inner_dim = 1; in Compute() local1367 int64 inner_dim = 1; in Compute() local1525 int64 inner_dim = 1; in Compute() local1816 int64 inner_dim = 1; in Compute() local2172 int64 inner_dim = 1; in Compute() local
39 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); in Compile() local
94 const int m1_rows, const int inner_dim, in multiply_mat()
927 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
1696 auto inner_dim = c->Dim(input, rank - 1); in __anonb22bfa861402() local
2445 [&](int64 product, int64 inner_dim) { in EmitFastConcatenate()
1204 int64 inner_dim = RandomDim(); in TEST_F() local