Home
last modified time | relevance | path

Searched refs:new_size_bounds (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h259 absl::Span<const int64> new_size_bounds,
Dshape_inference.cc2913 absl::Span<const int64> new_size_bounds, in InferDynamicReshapeShape() argument
2915 if (new_size_bounds.size() != dims_are_dynamic.size()) { in InferDynamicReshapeShape()
2919 new_size_bounds.size(), dims_are_dynamic.size()); in InferDynamicReshapeShape()
2931 operand.element_type(), new_size_bounds, dims_are_dynamic); in InferDynamicReshapeShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h479 absl::Span<const int64> new_size_bounds,
1081 absl::Span<const int64> new_size_bounds,
1695 absl::Span<const int64> new_size_bounds,
Dxla_builder.cc1149 absl::Span<const int64> new_size_bounds, in DynamicReshape() argument
1162 new_size_bounds, dims_are_dynamic)); in DynamicReshape()
4138 absl::Span<const int64> new_size_bounds, in DynamicReshape() argument
4140 return operand.builder()->DynamicReshape(operand, dim_sizes, new_size_bounds, in DynamicReshape()