Searched refs:new_size_bounds (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 259 absl::Span<const int64> new_size_bounds,
|
| D | shape_inference.cc | 2913 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/ |
| D | xla_builder.h | 479 absl::Span<const int64> new_size_bounds,
1081 absl::Span<const int64> new_size_bounds,
1695 absl::Span<const int64> new_size_bounds,
|
| D | xla_builder.cc | 1149 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()
|