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 << " 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()
106 const Tensor& backing_tensor = context->input(0); in Compute() local
108 OP_REQUIRES(context, backing_tensor.dtype() == dtype_, in Compute()
110 DataTypeString(backing_tensor.dtype()), in Compute()
115 OP_REQUIRES(context, backing_tensor.NumElements() >= shape_.num_elements(), in Compute()
[all …]
|
| 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()
|
| /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()
153 const Tensor& backing_tensor, int64 step_id, int32 scope_id, in AddScopedAllocator() argument
158 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,
84 const Tensor& backing_tensor, int64 step_id, int32 scope_id,
|
| D | scoped_allocator.cc | 22 ScopedAllocator::ScopedAllocator(const Tensor& backing_tensor, int32 scope_id, in ScopedAllocator() argument
27 : backing_tensor_(backing_tensor), in ScopedAllocator()
|
| D | scoped_allocator.h | 47 ScopedAllocator(const Tensor& backing_tensor, int32 scope_id,
|