Searched defs:new_sizes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 776 gtl::ArraySlice<int64> new_sizes, in FillDimsAndValidateCompatibleShape()
789 gtl::ArraySlice<int64> new_sizes, in FillDimsAndValidateCompatibleShape()
813 gtl::ArraySlice<int64> new_sizes) { in shaped()
822 gtl::ArraySlice<int64> new_sizes) { in bit_casted_shaped()
831 gtl::ArraySlice<int64> new_sizes) { in unaligned_shaped()
840 gtl::ArraySlice<int64> new_sizes) const { in shaped()
850 gtl::ArraySlice<int64> new_sizes) const { in bit_casted_shaped()
859 gtl::ArraySlice<int64> new_sizes) const { in unaligned_shaped()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op.cc | 48 std::vector<int64> new_sizes{1, input.NumElements(), 1}; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 395 def reshape(x, new_sizes, dimensions=None, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 892 absl::Span<const int64> new_sizes) { in Reshape()
906 absl::Span<const int64> new_sizes) { in Reshape()
940 std::vector<int64> new_sizes; in Collapse() local
2856 absl::Span<const int64> new_sizes) { in Reshape()
2860 XlaOp Reshape(const XlaOp& operand, absl::Span<const int64> new_sizes) { in Reshape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 489 absl::Span<const int64> new_sizes) { in Reshape()
|
| D | xla_client.py | 1188 def Reshape(self, operand, dimensions, new_sizes): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2679 absl::Span<const int64> new_sizes) { in InferReshapeShape()
|