| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 430 typename TTypes<T, NDIMS>::Tensor shaped(gtl::ArraySlice<int64> new_sizes);
439 gtl::ArraySlice<int64> new_sizes);
443 gtl::ArraySlice<int64> new_sizes);
497 gtl::ArraySlice<int64> new_sizes) const;
506 gtl::ArraySlice<int64> new_sizes) const;
510 gtl::ArraySlice<int64> new_sizes) const;
663 gtl::ArraySlice<int64> new_sizes,
668 gtl::ArraySlice<int64> new_sizes,
776 gtl::ArraySlice<int64> new_sizes, in FillDimsAndValidateCompatibleShape() argument
778 CHECK_EQ(NDIMS, new_sizes.size()); in FillDimsAndValidateCompatibleShape()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op.cc | 48 std::vector<int64> new_sizes{1, input.NumElements(), 1}; in Compute() local
84 new_sizes[0] *= input.dim_size(i); in Compute()
87 new_sizes[1] = input.dim_size(axis); in Compute()
90 new_sizes[2] *= input.dim_size(i); in Compute()
98 1, TensorShape({new_sizes[1]}), &idx)); in Compute()
102 if (new_sizes[0] == 1 && new_sizes[2] == 1) { in Compute()
132 auto Tin = input.shaped<T, 3>(new_sizes); in Compute()
170 new_sizes[1] = uniq_size; in Compute()
176 auto Tout = output->shaped<T, 3>(new_sizes); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 395 def reshape(x, new_sizes, dimensions=None, name=None): argument
398 x = array_ops.reshape(x, new_sizes, name=name)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 892 absl::Span<const int64> new_sizes) { in Reshape() argument
897 operand_shape, dimensions, new_sizes)); in Reshape()
906 absl::Span<const int64> new_sizes) { in Reshape() argument
911 return Reshape(operand, dimensions, new_sizes); in Reshape()
940 std::vector<int64> new_sizes; in Collapse() local
943 new_sizes.push_back(original_shape.dimensions(i)); in Collapse()
945 new_sizes.back() *= original_shape.dimensions(i); in Collapse()
949 VLOG(3) << "new sizes: [" << absl::StrJoin(new_sizes, ",") << "]"; in Collapse()
951 return Reshape(operand, new_sizes); in Collapse()
2856 absl::Span<const int64> new_sizes) { in Reshape() argument
[all …]
|
| D | xla_builder.h | 340 absl::Span<const int64> new_sizes);
342 XlaOp Reshape(const XlaOp& operand, absl::Span<const int64> new_sizes);
726 absl::Span<const int64> new_sizes);
728 friend XlaOp Reshape(const XlaOp& operand, absl::Span<const int64> new_sizes);
1153 absl::Span<const int64> new_sizes);
1158 XlaOp Reshape(const XlaOp& operand, absl::Span<const int64> new_sizes);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 231 absl::Span<const int64> new_sizes);
|
| D | shape_inference.cc | 2679 absl::Span<const int64> new_sizes) { in InferReshapeShape() argument
2683 ShapeUtil::MakeShape(operand.element_type(), new_sizes); in InferReshapeShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 489 absl::Span<const int64> new_sizes) { in Reshape() argument
490 return xla::Reshape(operand.op(), dimensions, new_sizes); in Reshape()
|
| D | local_computation_builder.h | 251 absl::Span<const int64> new_sizes);
|
| D | xla_client.py | 1188 def Reshape(self, operand, dimensions, new_sizes): argument
1203 return self._client.Reshape(operand, dimensions, new_sizes)
|
| D | xla_client_test.py | 1004 new_sizes=[2, 3])
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1968 <b> `Reshape(operand, new_sizes)` </b>
1969 <b> `Reshape(operand, dimensions, new_sizes)` </b>
1975 `new_sizes` | `int64` vector | vector of sizes of new dimensions
1985 input array into a single dimension. The `new_sizes` vector determines the size
1986 of the output array. The value at index 0 in `new_sizes` is the size of
1990 array defined by `new_sizes`, the dimensions in `new_sizes` are ordered from
|