Searched refs:backing_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 << backing_tensor->TotalBytes() << " num_elements_ " in Compute()
63 << num_elements_ << " buffer " << DMAHelper::buffer(backing_tensor) in Compute()
64 << " base addr " << DMAHelper::base(backing_tensor); in Compute()
66 s = sam->AddScopedAllocator(*backing_tensor, context->step_id(), id_, in Compute()
105 const Tensor& backing_tensor = context->input(0); in Compute() local
107 OP_REQUIRES(context, backing_tensor.dtype() == dtype_, in Compute()
109 DataTypeString(backing_tensor.dtype()), in Compute()
114 OP_REQUIRES(context, backing_tensor.NumElements() >= shape_.num_elements(), in Compute()
[all …]
|
| D | scoped_allocator_ops_test.cc | 86 Tensor** backing_tensor, Allocator* allocator, in PrepOp() argument
99 *backing_tensor = new Tensor(allocator, dtype, {num_elements}); in PrepOp()
101 Status s = sam->AddScopedAllocator(**backing_tensor, step_id, id, in PrepOp()
118 inputs->push_back({nullptr, *backing_tensor}); in PrepOp()
169 Tensor* backing_tensor = nullptr; in ExecOp() local
172 PrepOp(dtype, id, fields_shapes, &fields, &backing_tensor, allocator(), in ExecOp()
195 delete backing_tensor; in ExecOp()
288 Tensor* backing_tensor = nullptr; in ExecOp() local
291 PrepOp(dtype, id, fields_shapes, &fields, &backing_tensor, allocator(), in ExecOp()
315 delete backing_tensor; in ExecOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr.cc | 23 const Tensor& backing_tensor, int32 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 step_id, int32 scope_id, in AddScopedAllocator() argument
159 return sac->AddScopedAllocator(backing_tensor, scope_id, scope_name, fields, in AddScopedAllocator()
|
| D | scoped_allocator_mgr.h | 35 const Tensor& backing_tensor, int32 scope_id, const string& scope_name,
83 const Tensor& backing_tensor, int64 step_id, int32 scope_id,
|
| D | scoped_allocator.cc | 20 ScopedAllocator::ScopedAllocator(const Tensor& backing_tensor, int32 scope_id, in ScopedAllocator() argument
25 : backing_tensor_(backing_tensor), in ScopedAllocator()
|
| D | scoped_allocator.h | 46 ScopedAllocator(const Tensor& backing_tensor, int32 scope_id,
|