Searched refs:input_dims_only_sizes (Results 1 – 1 of 1) sorted by relevance
206 std::vector<int> input_dims_only_sizes; in ImportShape() local216 input_dims_only_sizes.push_back(d.size()); in ImportShape()236 *shape->mutable_dims() = input_dims_only_sizes; in ImportShape()240 return NumElements(input_dims_only_sizes, input_flat_size); in ImportShape()