Searched refs:dim_size_shapes (Results 1 – 3 of 3) sorted by relevance
258 const Shape& operand, absl::Span<const Shape* const> dim_size_shapes,
2912 const Shape& operand, absl::Span<const Shape* const> dim_size_shapes, in InferDynamicReshapeShape() argument2922 for (const Shape* dim_size_shape : dim_size_shapes) { in InferDynamicReshapeShape()
1154 TF_ASSIGN_OR_RETURN(const auto& dim_size_shapes, in DynamicReshape()1157 absl::c_transform(dim_size_shapes, std::back_inserter(dim_size_shape_ptrs), in DynamicReshape()