Searched refs:size_for_added_dims (Results 1 – 2 of 2) sorted by relevance
576 std::vector<int> size_for_added_dims = {},
2683 std::vector<int> size_for_added_dims, in DynamicReshape() argument2696 int max_num_slices = std::max(slices.size(), size_for_added_dims.size()); in DynamicReshape()2702 if (i < size_for_added_dims.size() && size_for_added_dims[i] >= 0) { in DynamicReshape()2704 CreateScalarConstant(params, size_for_added_dims[i], &tensor)); in DynamicReshape()