Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc147 XRTTupleAllocation::XRTTupleAllocation(int device_ordinal, in XRTTupleAllocation() function in tensorflow::XRTTupleAllocation
158 XRTTupleAllocation::~XRTTupleAllocation() { ReleaseBuffers(); } in ~XRTTupleAllocation()
160 void XRTTupleAllocation::ReleaseBuffers() { in ReleaseBuffers()
169 /*static*/ Status XRTTupleAllocation::CreateAndTransfer( in CreateAndTransfer()
172 XRTTupleAllocation** allocation) { in CreateAndTransfer()
187 *allocation = new XRTTupleAllocation( in CreateAndTransfer()
197 /*static*/ Status XRTTupleAllocation::CreateUninitialized( in CreateUninitialized()
200 XRTTupleAllocation** allocation) { in CreateUninitialized()
210 *allocation = new XRTTupleAllocation( in CreateUninitialized()
220 /*static*/ Status XRTTupleAllocation::CreateFromBuffer( in CreateFromBuffer()
[all …]
Dxrt_state.h76 class XRTTupleAllocation : public core::RefCounted {
78 ~XRTTupleAllocation() override;
86 XRTTupleAllocation** allocation);
94 XRTTupleAllocation** allocation);
99 XRTTupleAllocation** allocation);
109 XRTTupleAllocation** allocation);
116 static Status MakeSubBuffer(XRTTupleAllocation* parent,
118 XRTTupleAllocation** allocation,
125 RefPtr<XRTTupleAllocation> allocation;
142 XRTTupleAllocation** allocation);
[all …]
Dxrt_util.h77 xla::StatusOr<std::vector<RefPtr<XRTTupleAllocation>>> GetInputTupleAllocations(
84 const RefPtr<XRTTupleAllocation>& output_tuple,
85 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples,
90 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples,
98 RefPtr<XRTTupleAllocation> output_tuple,
104 std::function<xla::StatusOr<RefPtr<XRTTupleAllocation>>(
106 absl::Span<const RefPtr<XRTTupleAllocation>>)>;
Dxrt_util.cc78 Status Add(size_t index, RefPtr<XRTTupleAllocation> tuple) { in Add()
104 xla::StatusOr<RefPtr<XRTTupleAllocation>> Lookup(size_t index) const { in Lookup()
136 Status MakeOutput(const RefPtr<XRTTupleAllocation>& output, int64 index, in MakeOutput()
137 RefPtr<XRTTupleAllocation>* result) { in MakeOutput()
141 XRTTupleAllocation* tuple; in MakeOutput()
143 XRTTupleAllocation::MakeSubBuffer(output.get(), {index - 1}, &tuple, in MakeOutput()
255 xla::StatusOr<std::vector<RefPtr<XRTTupleAllocation>>> GetInputTupleAllocations( in GetInputTupleAllocations()
265 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetInputTupleAllocations()
288 XRTTupleAllocation* sub_tuple; in GetInputTupleAllocations()
289 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in GetInputTupleAllocations()
[all …]
Dxrt_memory_manager.h61 const std::vector<RefPtr<XRTTupleAllocation>>& PinnedTuples() const { in PinnedTuples()
71 std::vector<RefPtr<XRTTupleAllocation>> pinned_tuples_;
79 int64 Register(RefPtr<XRTTupleAllocation> tuple);
83 xla::StatusOr<RefPtr<XRTTupleAllocation>> Lookup(int64 handle);
85 Status Lookup(int64 handle, RefPtr<XRTTupleAllocation>* tuple) { in Lookup()
Dxrt_memory_manager.cc50 explicit Alloc(RefPtr<XRTTupleAllocation> tuple) in Alloc()
53 RefPtr<XRTTupleAllocation> tuple;
59 int64 Register(RefPtr<XRTTupleAllocation> tuple) { in Register()
84 RefPtr<XRTTupleAllocation> Lookup(int64 handle) { in Lookup()
218 int64 XRTMemoryManager::Register(RefPtr<XRTTupleAllocation> tuple) { in Register()
224 xla::StatusOr<RefPtr<XRTTupleAllocation>> XRTMemoryManager::Lookup( in Lookup()
232 RefPtr<XRTTupleAllocation> tuple = device_context->Lookup(handle); in Lookup()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc88 xla::StatusOr<std::vector<RefPtr<XRTTupleAllocation>>> GetInputTuples( in GetInputTuples()
101 xla::StatusOr<std::vector<RefPtr<XRTTupleAllocation>>> GetChainedOpInputTuples( in GetChainedOpInputTuples()
103 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs) { in GetChainedOpInputTuples()
104 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetChainedOpInputTuples()
114 XRTTupleAllocation* sub_tuple; in GetChainedOpInputTuples()
115 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in GetChainedOpInputTuples()
267 xla::StatusOr<RefPtr<XRTTupleAllocation>> CreateOutputTuple( in CreateOutputTuple()
270 XRTTupleAllocation* output_tuple; in CreateOutputTuple()
281 TF_RETURN_IF_ERROR(XRTTupleAllocation::CreateFromBuffer( in CreateOutputTuple()
287 TF_RETURN_IF_ERROR(XRTTupleAllocation::CreateFromBuffer( in CreateOutputTuple()
[all …]
Dtpu_execute_op.cc85 xla::StatusOr<std::vector<RefPtr<XRTTupleAllocation>>> GetChainedOpInputs( in GetChainedOpInputs()
87 absl::Span<const RefPtr<XRTTupleAllocation>> op_inputs, in GetChainedOpInputs()
95 std::vector<RefPtr<XRTTupleAllocation>> input_tuples; in GetChainedOpInputs()
99 const RefPtr<XRTTupleAllocation>& tuple = op_inputs[i]; in GetChainedOpInputs()
106 XRTTupleAllocation* sub_tuple; in GetChainedOpInputs()
107 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in GetChainedOpInputs()
137 xla::StatusOr<RefPtr<XRTTupleAllocation>> AllocateOutputTuple( in AllocateOutputTuple()
139 absl::Span<const RefPtr<XRTTupleAllocation>> input_tuples, in AllocateOutputTuple()
151 XRTTupleAllocation* output_tuple; in AllocateOutputTuple()
155 TF_RETURN_IF_ERROR(XRTTupleAllocation::CreateFromBuffer( in AllocateOutputTuple()
[all …]
Dxrt_state_ops.h75 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleNode()
97 XRTTupleAllocation::ExpandedTupleInput& input = in ParseTupleNode()
127 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleTree()
128 xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>* tuple_shape_tree, in ParseTupleTree()
139 *tuple_shape_tree = xla::ShapeTree<XRTTupleAllocation::ExpandedTupleInput>( in ParseTupleTree()
143 XRTTupleAllocation::ExpandedTupleInput* element) { in ParseTupleTree()
201 XRTTupleAllocation* allocation; in Compute()
202 OP_REQUIRES_OK(ctx, XRTTupleAllocation::CreateAndTransfer( in Compute()
241 XRTTupleAllocation* allocation; in Compute()
243 XRTTupleAllocation::CreateUninitialized( in Compute()
[all …]