Searched defs:new_sizes (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | tensor.h | 804 gtl::ArraySlice<int64> new_sizes, in FillDimsAndValidateCompatibleShape() 817 gtl::ArraySlice<int64> new_sizes, in FillDimsAndValidateCompatibleShape() 841 gtl::ArraySlice<int64> new_sizes) { in shaped() 850 gtl::ArraySlice<int64> new_sizes) { in bit_casted_shaped() 859 gtl::ArraySlice<int64> new_sizes) { in unaligned_shaped() 868 gtl::ArraySlice<int64> new_sizes) const { in shaped() 878 gtl::ArraySlice<int64> new_sizes) const { in bit_casted_shaped() 887 gtl::ArraySlice<int64> new_sizes) const { in unaligned_shaped()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | unique_op.cc | 92 std::vector<int64> new_sizes{1, input.NumElements(), 1}; in Compute() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
D | xla.py | 419 def reshape(x, new_sizes, dimensions=None, name=None): argument
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 1116 absl::Span<const int64> new_sizes, in Reshape() 1130 XlaOp XlaBuilder::Reshape(XlaOp operand, absl::Span<const int64> new_sizes, in Reshape() 1201 std::vector<int64> new_sizes; in Collapse() local 4125 absl::Span<const int64> new_sizes) { in Reshape() 4129 XlaOp Reshape(const XlaOp operand, absl::Span<const int64> new_sizes) { in Reshape() 4145 absl::Span<const int64> new_sizes, in ReshapeWithInferredDimension()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference.cc | 2945 absl::Span<const int64> new_sizes, int64 inferred_dimension) { in InferReshapeShape()
|