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