Searched refs:LiteralBase (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 111 LiteralBase::~LiteralBase() {} in ~LiteralBase()
218 Literal LiteralBase::CreateFromShape(const Shape& shape) { in CreateFromShape()
229 int32 LiteralBase::GetDynamicSize(int64 dim_index) const { in GetDynamicSize()
233 int32 LiteralBase::GetDynamicSize(int64 dim_index, in GetDynamicSize()
238 absl::optional<int64> LiteralBase::GetFirstInteger() const { in GetFirstInteger()
268 const LiteralBase& src_literal, absl::Span<const int64> src_base, in CopySliceFromInternal()
452 int32 LiteralBase::Piece::GetDynamicSize(int64 dim_index) const { in GetDynamicSize()
462 void LiteralBase::Piece::SetDynamicSize(int64 dim_index, int32 size) { in SetDynamicSize()
477 Status LiteralBase::Piece::CopyFrom(const LiteralBase::Piece& src, in CopyFrom()
628 src_literal.root_piece_ = new LiteralBase::Piece(); in MoveFrom()
[all …]
|
| D | literal.h | 58 class LiteralBase {
60 virtual ~LiteralBase() = 0;
64 bool operator==(const LiteralBase& other) const;
65 bool operator!=(const LiteralBase& other) const { return !(*this == other); }
459 const_cast<xla::LiteralBase::Piece*>(this), &index) in ForEachMutableSubpiece()
469 func, const_cast<xla::LiteralBase::Piece*>(this), &index); in ForEachMutableSubpieceWithStatus()
544 void CopyElementsWithDynamicBound(const LiteralBase::Piece& src);
584 class MutableLiteralBase : public LiteralBase {
595 using LiteralBase::data;
611 using LiteralBase::untyped_data;
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.h | 31 const LiteralBase& literal, mlir::Builder builder);
|
| D | hlo_utils.cc | 36 using xla::LiteralBase;
41 const ShapedType& type, const LiteralBase& literal) { in CreateDenseAttrFromLiteral()
99 const LiteralBase& literal, Builder builder) { in CreateDenseElementsAttrFromLiteral()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 83 static Status CreateAndTransfer(const xla::LiteralBase& literal,
|
| D | xrt_state.cc | 170 const xla::LiteralBase& literal, XRTMemoryManager* memory_manager, in CreateAndTransfer()
|