Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.h32 class XrtAllocation {
37 static StatusOr<XrtAllocation*> FromLiteral(const Literal& argument,
40 XrtAllocation(int64 handle, Shape shape, const string& session_target);
41 ~XrtAllocation();
57 explicit XrtAllocationTuple(std::vector<XrtAllocation*> elements);
63 StatusOr<XrtAllocation*> Release(int i);
69 std::vector<XrtAllocation*> elements_;
79 XrtAllocation* allocation, const string& session_target);
99 StatusOr<XrtAllocation*> Execute(
100 absl::Span<XrtAllocation* const> argument_handles);
[all …]
Dxrt.cc47 XrtAllocation::XrtAllocation(int64 handle, Shape shape, in XrtAllocation() function in xla::swig::XrtAllocation
51 XrtAllocation::~XrtAllocation() { in ~XrtAllocation()
74 StatusOr<XrtAllocation*> XrtAllocation::FromLiteral( in FromLiteral()
92 return new XrtAllocation(handle, argument.shape(), session_target); in FromLiteral()
95 const int64 XrtAllocation::handle() const { return handle_; } in handle()
97 const Shape& XrtAllocation::shape() const { return shape_; } in shape()
99 StatusOr<Literal> XrtAllocation::ToLiteral() const { in ToLiteral()
117 XrtAllocationTuple::XrtAllocationTuple(std::vector<XrtAllocation*> elements) in XrtAllocationTuple()
125 for (XrtAllocation* element : elements_) { in ~XrtAllocationTuple()
132 StatusOr<XrtAllocation*> XrtAllocationTuple::Release(int i) { in Release()
[all …]
Dxrt.i53 %typemap(out) StatusOr<xla::swig::XrtAllocation*> {
58 $typemap(out, xla::swig::XrtAllocation*)
80 %typemap(in) absl::Span<xla::swig::XrtAllocation* const>
81 (std::vector<XrtAllocation*> temps) {
90 XrtAllocation* xrta;
91 if ((SWIG_ConvertPtr(o, (void**) &xrta, $descriptor(xla::swig::XrtAllocation*),
105 %unignore xla::swig::XrtAllocation;
106 %unignore xla::swig::XrtAllocation::FromLiteral;
107 %unignore xla::swig::XrtAllocation::ToLiteral;
108 %unignore xla::swig::XrtAllocation::shape;
Dxla_client.py173 return xrt_api.XrtAllocation.FromLiteral(pyval,