Searched defs:new_sizes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 824 gtl::ArraySlice<int64_t> new_sizes, in FillDimsAndValidateCompatibleShape()
837 gtl::ArraySlice<int64_t> new_sizes, in FillDimsAndValidateCompatibleShape()
880 gtl::ArraySlice<int64_t> new_sizes) { in shaped()
889 gtl::ArraySlice<int64_t> new_sizes) { in bit_casted_shaped()
898 gtl::ArraySlice<int64_t> new_sizes) { in unaligned_shaped()
907 gtl::ArraySlice<int64_t> new_sizes) const { in shaped()
917 gtl::ArraySlice<int64_t> new_sizes) const { in bit_casted_shaped()
926 gtl::ArraySlice<int64_t> new_sizes) const { in unaligned_shaped()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op.cc | 92 std::vector<int64_t> new_sizes{1, input.NumElements(), 1}; in Compute() local
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | utils.cc | 158 SmallVector<int32_t> new_sizes = llvm::to_vector(values); in SizedOperandSegmentsEraseOperands() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 493 def reshape(x, new_sizes, dimensions=None, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1183 absl::Span<const int64_t> new_sizes, in Reshape()
1197 XlaOp XlaBuilder::Reshape(XlaOp operand, absl::Span<const int64_t> new_sizes, in Reshape()
1270 std::vector<int64_t> new_sizes; in Collapse() local
4081 absl::Span<const int64_t> new_sizes) { in Reshape()
4085 XlaOp Reshape(const XlaOp operand, absl::Span<const int64_t> new_sizes) { in Reshape()
4101 absl::Span<const int64_t> new_sizes, in ReshapeWithInferredDimension()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | tensor_layout.cc | 141 std::vector<int> new_sizes; in operator ==() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 3085 absl::Span<const int64_t> new_sizes, int64_t inferred_dimension) { in InferReshapeShape()
|