Searched refs:element_shape_vec (Results 1 – 1 of 1) sorted by relevance
110 const auto element_shape_vec = element_shape->vec<int64>(); in Compute() local117 output_shape_vec(j + input_dims_to_stack) = element_shape_vec(j); in Compute()144 const auto element_shape_vec = element_shape->vec<int64>(); in Compute() local147 output_shape_vec(j + input_dims_to_stack), element_shape_vec(j)); in Compute()