Home
last modified time | relevance | path

Searched refs:LiteralBase (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc77 LiteralBase::~LiteralBase() {} in ~LiteralBase()
187 Literal LiteralBase::CreateFromShape(const Shape& shape) { in CreateFromShape()
198 const SparseIndexArray* LiteralBase::sparse_indices( in sparse_indices()
210 const LiteralBase& src_literal, absl::Span<const int64> src_base, in CopySliceFromInternal()
387 Status LiteralBase::Piece::CopyFrom(const LiteralBase::Piece& src) { in CopyFrom()
498 src_literal.root_piece_ = new LiteralBase::Piece(); in MoveFrom()
578 Literal LiteralBase::Relayout(const Layout& new_layout, in Relayout()
590 Literal LiteralBase::Relayout(const Shape& shape_with_layout) const { in Relayout()
608 StatusOr<Literal> LiteralBase::Broadcast( in Broadcast()
648 StatusOr<Literal> LiteralBase::Reshape( in Reshape()
[all …]
Dliteral.h58 class LiteralBase {
60 virtual ~LiteralBase() = 0;
64 bool operator==(const LiteralBase& other) const;
65 bool operator!=(const LiteralBase& other) const { return !(*this == other); }
426 const_cast<xla::LiteralBase::Piece*>(this), &index) in ForEachMutableSubpiece()
436 func, const_cast<xla::LiteralBase::Piece*>(this), &index); in ForEachMutableSubpieceWithStatus()
550 class MutableLiteralBase : public LiteralBase {
561 using LiteralBase::data;
576 using LiteralBase::untyped_data;
698 return const_cast<Piece&>(LiteralBase::piece(shape_index)); in piece()
[all …]
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.h91 static Status CreateAndTransfer(const xla::LiteralBase& literal,
Dxrt_state.cc184 const xla::LiteralBase& literal, xla::Backend* backend, int device_ordinal, in CreateAndTransfer()