Searched refs:out_shape_flat (Results 1 – 2 of 2) sorted by relevance
321 auto out_shape_flat = out_shape_t->flat<int64>(); in Compute() local323 std::copy(out_dim_sizes.begin(), out_dim_sizes.end(), &out_shape_flat(0)); in Compute()
183 auto out_shape_flat = out_shape_t->vec<int64>(); in OutputSparseTensor() local185 out_shape_flat(i) = output_shape.dim_size(i); in OutputSparseTensor()