Searched refs:out_temp (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 237 Tensor* out_temp) { in allocate_temp() argument
250 *out_temp = new_temp; in allocate_temp()
567 Tensor* out_temp) { in forward_input_or_allocate_temp() argument
573 *out_temp = std::move(*new_tensor); in forward_input_or_allocate_temp()
577 return allocate_temp(type, shape, out_temp, allocator_attr); in forward_input_or_allocate_temp()
743 DataType type, const TensorShape& shape, Tensor* out_temp, in allocate_temp() argument
747 allocate_tensor(type, shape, out_temp, allocator_attr, allocation_attr); in allocate_temp()
748 if (track_allocations() && s.ok() && out_temp->TotalBytes() > 0) { in allocate_temp()
751 int64 alloc_size = a->AllocatedSize(out_temp->tensor_data().data()); in allocate_temp()
752 record_temp_memory_allocation(alloc_size, *out_temp); in allocate_temp()
|
| D | op_kernel.h | 295 Tensor* out_temp);
872 Tensor* out_temp) TF_MUST_USE_RESULT;
876 const TensorShape& shape, Tensor* out_temp) TF_MUST_USE_RESULT { in forward_input_or_allocate_temp() argument
878 AllocatorAttributes(), out_temp); in forward_input_or_allocate_temp()
968 Tensor* out_temp, AllocatorAttributes allocator_attr,
971 Tensor* out_temp, AllocatorAttributes allocator_attr) { in allocate_temp() argument
972 return allocate_temp(type, shape, out_temp, allocator_attr, in allocate_temp()
976 Tensor* out_temp) { in allocate_temp() argument
977 return allocate_temp(type, shape, out_temp, AllocatorAttributes()); in allocate_temp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1404 SparseBuffer out_temp; in FastParseSingleExample() local
1446 if (!feature.ParseInt64List(&out_temp.int64_list)) in FastParseSingleExample()
1448 num_elements = out_temp.int64_list.size(); in FastParseSingleExample()
1454 if (!feature.ParseFloatList(&out_temp.float_list)) in FastParseSingleExample()
1456 num_elements = out_temp.float_list.size(); in FastParseSingleExample()
1464 out_temp.bytes_list.reserve(actual_num_elements); in FastParseSingleExample()
1465 if (!feature.ParseBytesList(&out_temp.bytes_list)) in FastParseSingleExample()
1467 num_elements = out_temp.bytes_list.size(); in FastParseSingleExample()
1517 CopyOrMoveBlock(out_temp.int64_list.begin(), in FastParseSingleExample()
1518 out_temp.int64_list.end(), out->flat<int64>().data()); in FastParseSingleExample()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 275 Tensor* out_temp) { in allocate_scoped_tensor() argument
276 const Status status = context_->allocate_temp(type, shape, out_temp); in allocate_scoped_tensor()
278 scratch_tensor_refs_.emplace_back(*out_temp); in allocate_scoped_tensor()
285 const TensorShape& shape, Tensor* out_temp) { in forward_input_or_allocate_scoped_tensor() argument
287 candidate_input_indices, type, shape, out_temp); in forward_input_or_allocate_scoped_tensor()
289 scratch_tensor_refs_.emplace_back(*out_temp); in forward_input_or_allocate_scoped_tensor()
|