Home
last modified time | relevance | path

Searched defs:new_sizes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor.h804 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/
Dunique_op.cc92 std::vector<int64> new_sizes{1, input.NumElements(), 1}; in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py419 def reshape(x, new_sizes, dimensions=None, name=None): argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1116 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/
Dshape_inference.cc2945 absl::Span<const int64> new_sizes, int64 inferred_dimension) { in InferReshapeShape()