Searched refs:src_literal (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 210 const LiteralBase& src_literal, absl::Span<const int64> src_base, in CopySliceFromInternal() argument
212 TF_RET_CHECK(src_literal.shape().rank() == src_base.size()); in CopySliceFromInternal()
220 if (src_literal.shape().rank() == 0 || shape().rank() == 0) { in CopySliceFromInternal()
225 src_literal.data<NativeT>(), in CopySliceFromInternal()
226 linear_index(src_literal.shape(), src_base), 0, 1); in CopySliceFromInternal()
228 !ShapeUtil::IsZeroElementArray(src_literal.shape())) { in CopySliceFromInternal()
240 MutableLiteralBase::StrideConfig stride_config(src_literal.shape(), shape(), in CopySliceFromInternal()
251 int64 src_index = linear_index(src_literal.shape(), src_indexes); in CopySliceFromInternal()
256 src_literal.data<NativeT>(), src_index, in CopySliceFromInternal()
261 ShapeUtil::ForEachIndex(src_literal.shape(), stride_config.base, in CopySliceFromInternal()
[all …]
|
| D | literal.h | 592 Status CopyFrom(const LiteralSlice& src_literal,
606 Status CopySliceFrom(const LiteralSlice& src_literal,
612 Status CopyElementFrom(const LiteralSlice& src_literal,
706 Status CopySliceFromInternal(const LiteralBase& src_literal,
776 virtual Status MoveFrom(Literal&& src_literal,
|