Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc167 XRTTupleAllocation::XRTTupleAllocation(int device_ordinal, in XRTTupleAllocation() function in tensorflow::XRTTupleAllocation
177 XRTTupleAllocation::~XRTTupleAllocation() { in ~XRTTupleAllocation()
183 /*static*/ Status XRTTupleAllocation::CreateAndTransfer( in CreateAndTransfer()
185 XRTTupleAllocation** allocation) { in CreateAndTransfer()
201 *allocation = new XRTTupleAllocation(device_ordinal, allocator, in CreateAndTransfer()
209 /*static*/ Status XRTTupleAllocation::CreateFromBuffer( in CreateFromBuffer()
211 int device_ordinal, XRTTupleAllocation** allocation) { in CreateFromBuffer()
214 *allocation = new XRTTupleAllocation(device_ordinal, allocator, in CreateFromBuffer()
222 Status XRTTupleAllocation::ToLiteral(xla::Backend* backend, int device_ordinal, in ToLiteral()
241 Status XRTTupleAllocation::WriteLiteral(xla::Backend* backend, in WriteLiteral()
[all …]
Dxrt_state.h84 class XRTTupleAllocation : public ResourceBase {
86 ~XRTTupleAllocation() override;
93 XRTTupleAllocation** allocation);
98 XRTTupleAllocation** allocation);
105 static Status MakeSubBuffer(XRTTupleAllocation* parent,
107 XRTTupleAllocation** allocation,
114 XRTTupleAllocation* allocation;
130 XRTTupleAllocation** allocation);
135 XRTTupleAllocation** allocation);
181 Status AliasBufferFrom(const XRTTupleAllocation& source,
[all …]
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h71 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleNode()
93 XRTTupleAllocation::ExpandedTupleInput& input = in ParseTupleNode()
107 XRTTupleAllocation::Lookup(rm, key, &input.allocation)); in ParseTupleNode()
123 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleTree()
124 xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>* tuple_shape_tree, in ParseTupleTree()
135 *tuple_shape_tree = xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>( in ParseTupleTree()
139 XRTTupleAllocation::ExpandedTupleInput* element) { in ParseTupleTree()
195 XRTTupleAllocation* allocation; in Compute()
196 OP_REQUIRES_OK(ctx, XRTTupleAllocation::CreateAndTransfer( in Compute()
294 XRTTupleAllocation* allocation; in Compute()
[all …]
Dxrt_execute_op.cc70 std::vector<XRTTupleAllocation*>* input_tuples, in GetComputationInputs()
98 XRTTupleAllocation::Lookup(rm, input_uid, &(*input_tuples)[i])); in GetComputationInputs()
103 XRTTupleAllocation::DeleteFromResourceManager(rm, input_uid)); in GetComputationInputs()
106 XRTTupleAllocation* tuple = (*input_tuples)[i]; in GetComputationInputs()
174 std::vector<XRTTupleAllocation*> input_tuples; in DoWork()
227 XRTTupleAllocation* output_tuple; in DoWork()
228 TF_RETURN_IF_ERROR(XRTTupleAllocation::CreateFromBuffer( in DoWork()
262 XRTTupleAllocation* suballocation; in DoWork()
263 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in DoWork()