Home
last modified time | relevance | path

Searched refs:new_sizes (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor.h493 typename TTypes<T, NDIMS>::Tensor shaped(gtl::ArraySlice<int64> new_sizes);
502 gtl::ArraySlice<int64> new_sizes);
506 gtl::ArraySlice<int64> new_sizes);
560 gtl::ArraySlice<int64> new_sizes) const;
569 gtl::ArraySlice<int64> new_sizes) const;
573 gtl::ArraySlice<int64> new_sizes) const;
725 gtl::ArraySlice<int64> new_sizes,
730 gtl::ArraySlice<int64> new_sizes,
804 gtl::ArraySlice<int64> new_sizes, in FillDimsAndValidateCompatibleShape() argument
806 CHECK_EQ(NDIMS, new_sizes.size()); in FillDimsAndValidateCompatibleShape()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dunique_op.cc92 std::vector<int64> new_sizes{1, input.NumElements(), 1}; in Compute() local
128 new_sizes[0] *= input.dim_size(i); in Compute()
131 new_sizes[1] = input.dim_size(axis); in Compute()
134 new_sizes[2] *= input.dim_size(i); in Compute()
142 1, TensorShape({new_sizes[1]}), &idx)); in Compute()
146 if (new_sizes[0] == 1 && new_sizes[2] == 1) { in Compute()
176 auto Tin = input.shaped<T, 3>(new_sizes); in Compute()
215 new_sizes[1] = uniq_size; in Compute()
221 auto Tout = output->shaped<T, 3>(new_sizes); in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py419 def reshape(x, new_sizes, dimensions=None, name=None): argument
422 x = array_ops.reshape(x, new_sizes, name=name)
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1116 absl::Span<const int64> new_sizes, in Reshape() argument
1122 new_sizes, inferred_dimension)); in Reshape()
1130 XlaOp XlaBuilder::Reshape(XlaOp operand, absl::Span<const int64> new_sizes, in Reshape() argument
1136 return Reshape(operand, dimensions, new_sizes, inferred_dimension); in Reshape()
1201 std::vector<int64> new_sizes; in Collapse() local
1204 new_sizes.push_back(original_shape->dimensions(i)); in Collapse()
1206 new_sizes.back() *= original_shape->dimensions(i); in Collapse()
1210 VLOG(3) << "new sizes: [" << absl::StrJoin(new_sizes, ",") << "]"; in Collapse()
1212 return Reshape(operand, new_sizes); in Collapse()
4125 absl::Span<const int64> new_sizes) { in Reshape() argument
[all …]
Dxla_builder.h469 absl::Span<const int64> new_sizes,
472 XlaOp Reshape(XlaOp operand, absl::Span<const int64> new_sizes,
1074 absl::Span<const int64> new_sizes);
1076 friend XlaOp Reshape(XlaOp operand, absl::Span<const int64> new_sizes);
1085 absl::Span<const int64> new_sizes,
1689 absl::Span<const int64> new_sizes);
1701 XlaOp Reshape(XlaOp operand, absl::Span<const int64> new_sizes);
1711 absl::Span<const int64> new_sizes,
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h250 absl::Span<const int64> new_sizes,
Dshape_inference.cc2945 absl::Span<const int64> new_sizes, int64 inferred_dimension) { in InferReshapeShape() argument
2949 ShapeUtil::MakeShape(operand.element_type(), new_sizes); in InferReshapeShape()
2974 auto common_factors = CommonFactors(operand.dimensions(), new_sizes); in InferReshapeShape()
3046 if (operand.dimensions(input_dim) == 1 && !new_sizes.empty()) { in InferReshapeShape()
3053 output_dynamic_dimension = new_sizes.size() - 1; in InferReshapeShape()
3081 if (new_sizes[i] != 1) { in InferReshapeShape()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2212 <b> `Reshape(operand, new_sizes)` </b>
2213 <b> `Reshape(operand, dimensions, new_sizes)` </b>
2219 `new_sizes` | `int64` vector | vector of sizes of new dimensions
2229 input array into a single dimension. The `new_sizes` vector determines the size
2230 of the output array. The value at index 0 in `new_sizes` is the size of
2234 array defined by `new_sizes`, the dimensions in `new_sizes` are ordered from
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py1078 new_sizes=[2, 3])