Home
last modified time | relevance | path

Searched refs:backing_tensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dscoped_allocator_ops.cc57 Tensor* backing_tensor = nullptr; in Compute() local
60 context->allocate_output(0, {num_elements_}, &backing_tensor, attr); in Compute()
62 << " new backing tensor size " << backing_tensor->TotalBytes() in Compute()
64 << DMAHelper::buffer(backing_tensor) << " base addr " in Compute()
65 << DMAHelper::base(backing_tensor); in Compute()
67 s = sam->AddScopedAllocator(*backing_tensor, context->step_id(), id_, in Compute()
109 const Tensor& backing_tensor = context->input(0); in Compute() local
111 OP_REQUIRES(context, backing_tensor.dtype() == dtype_, in Compute()
113 DataTypeString(backing_tensor.dtype()), in Compute()
118 OP_REQUIRES(context, backing_tensor.NumElements() >= shape_.num_elements(), in Compute()
[all …]
H A Dscoped_allocator_ops_test.cc87 Tensor** backing_tensor, Allocator* allocator, in PrepOp() argument
100 *backing_tensor = new Tensor(allocator, dtype, {num_elements}); in PrepOp()
102 Status s = sam->AddScopedAllocator(**backing_tensor, step_id, id, in PrepOp()
119 inputs->push_back({nullptr, *backing_tensor}); in PrepOp()
170 Tensor* backing_tensor = nullptr; in ExecOp() local
173 PrepOp(dtype, id, fields_shapes, &fields, &backing_tensor, allocator(), in ExecOp()
196 delete backing_tensor; in ExecOp()
289 Tensor* backing_tensor = nullptr; in ExecOp() local
292 PrepOp(dtype, id, fields_shapes, &fields, &backing_tensor, allocator(), in ExecOp()
316 delete backing_tensor; in ExecOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dscoped_allocator_mgr.cc23 const Tensor& backing_tensor, int32_t scope_id, const string& scope_name, in AddScopedAllocator() argument
45 backing_tensor, scope_id, scope_name, fields, expected_call_count, this); in AddScopedAllocator()
154 const Tensor& backing_tensor, int64_t step_id, int32_t scope_id, in AddScopedAllocator() argument
159 return sac->AddScopedAllocator(backing_tensor, scope_id, scope_name, fields, in AddScopedAllocator()
H A Dscoped_allocator_mgr.h35 const Tensor& backing_tensor, int32_t scope_id,
84 const Tensor& backing_tensor, int64_t step_id, int32_t scope_id,
H A Dscoped_allocator.cc22 ScopedAllocator::ScopedAllocator(const Tensor& backing_tensor, int32_t scope_id, in ScopedAllocator() argument
27 : backing_tensor_(backing_tensor), in ScopedAllocator()
H A Dscoped_allocator.h47 ScopedAllocator(const Tensor& backing_tensor, int32_t scope_id,
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DFunctionHelpers.h64 typename TargetInfo::TensorType *backing_tensor = nullptr; in get_backing_tensor() local
71 …backing_tensor = (tensor_handle != nullptr) ? arm_compute::utils::cast::polymorphic_cast<typename … in get_backing_tensor()
74 return backing_tensor; in get_backing_tensor()