Searched refs:backing_tensor (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | scoped_allocator_ops.cc | 57 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 D | scoped_allocator_ops_test.cc | 87 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 D | scoped_allocator_mgr.cc | 23 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 D | scoped_allocator_mgr.h | 35 const Tensor& backing_tensor, int32_t scope_id, 84 const Tensor& backing_tensor, int64_t step_id, int32_t scope_id,
|
H A D | scoped_allocator.cc | 22 ScopedAllocator::ScopedAllocator(const Tensor& backing_tensor, int32_t scope_id, in ScopedAllocator() argument 27 : backing_tensor_(backing_tensor), in ScopedAllocator()
|
H A D | scoped_allocator.h | 47 ScopedAllocator(const Tensor& backing_tensor, int32_t scope_id,
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/ |
H A D | FunctionHelpers.h | 64 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()
|