Searched refs:out_temp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.cc | 188 Tensor* out_temp) { in allocate_scoped_tensor() argument
189 const Status status = context_->allocate_temp(type, shape, out_temp); in allocate_scoped_tensor()
191 scratch_tensor_refs_.emplace_back(*out_temp); in allocate_scoped_tensor()
198 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor() argument
200 candidate_input_indices, type, shape, out_temp); in forward_input_or_allocate_scoped_tensor()
202 scratch_tensor_refs_.emplace_back(*out_temp); in forward_input_or_allocate_scoped_tensor()
|
| D | cuda_solvers.cc | 296 Tensor* out_temp) { in allocate_scoped_tensor() argument
297 const Status status = context_->allocate_temp(type, shape, out_temp); in allocate_scoped_tensor()
299 scratch_tensor_refs_.emplace_back(*out_temp); in allocate_scoped_tensor()
306 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor() argument
308 candidate_input_indices, type, shape, out_temp); in forward_input_or_allocate_scoped_tensor()
310 scratch_tensor_refs_.emplace_back(*out_temp); in forward_input_or_allocate_scoped_tensor()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 287 Tensor* out_temp);
289 Tensor* out_temp, AllocatorAttributes allocator_attr);
917 Tensor* out_temp) TF_MUST_USE_RESULT;
921 const TensorShape& shape, Tensor* out_temp) TF_MUST_USE_RESULT { in forward_input_or_allocate_temp() argument
923 AllocatorAttributes(), out_temp); in forward_input_or_allocate_temp()
1012 Tensor* out_temp, AllocatorAttributes allocator_attr,
1015 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp() argument
1016 return allocate_temp(type, shape, out_temp, allocator_attr, in allocate_temp()
1020 Tensor* out_temp) { in allocate_temp() argument
1021 return allocate_temp(type, shape, out_temp, AllocatorAttributes()); in allocate_temp()
|
| D | op_kernel.cc | 262 Tensor* out_temp) { in allocate_temp() argument
275 *out_temp = new_temp; in allocate_temp()
281 Tensor* out_temp, in allocate_temp() argument
300 *out_temp = new_temp; in allocate_temp()
569 Tensor* out_temp) { in forward_input_or_allocate_temp() argument
575 *out_temp = std::move(*new_tensor); in forward_input_or_allocate_temp()
579 return allocate_temp(type, shape, out_temp, allocator_attr); in forward_input_or_allocate_temp()
776 DataType type, const TensorShape& shape, Tensor* out_temp, in allocate_temp() argument
798 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
799 if (track_allocations() && s.ok() && out_temp->TotalBytes() > 0) { in allocate_temp()
[all …]
|