Searched refs:dest_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 85 const Shape& source_shape, const Shape& dest_shape, in StrideConfig() argument
94 dimensions[LayoutUtil::Minor(dest_shape.layout(), 0)]) { in StrideConfig()
96 dest_stride = IndexUtil::GetDimensionStride(dest_shape, minor_dimension); in StrideConfig()
98 minor_dimension = LayoutUtil::Minor(dest_shape.layout(), 0); in StrideConfig()
372 absl::Span<const NativeT> src, const Shape& dest_shape, in CopyElementsBetween() argument
374 CHECK(ShapeUtil::Compatible(dest_shape, src_shape)); in CopyElementsBetween()
375 if (ShapeUtil::IsZeroElementArray(dest_shape)) { in CopyElementsBetween()
378 std::vector<int64> index(dest_shape.rank()); in CopyElementsBetween()
380 dest[IndexUtil::MultidimensionalIndexToLinearIndex(dest_shape, index)] = in CopyElementsBetween()
382 } while (IndexUtil::BumpIndices(dest_shape, absl::MakeSpan(index))); in CopyElementsBetween()
[all …]
|
| D | literal.h | 235 StatusOr<Literal> ConvertToShape(const Shape& dest_shape) const;
714 StrideConfig(const Shape& source_shape, const Shape& dest_shape,
|