/external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/ |
D | ignite_binary_object_parser.cc | 35 out_tensors->emplace_back(cpu_allocator(), DT_UINT8, TensorShape({})); in Parse() 40 out_tensors->emplace_back(cpu_allocator(), DT_INT16, TensorShape({})); in Parse() 45 out_tensors->emplace_back(cpu_allocator(), DT_UINT16, TensorShape({})); in Parse() 50 out_tensors->emplace_back(cpu_allocator(), DT_INT32, TensorShape({})); in Parse() 55 out_tensors->emplace_back(cpu_allocator(), DT_INT64, TensorShape({})); in Parse() 60 out_tensors->emplace_back(cpu_allocator(), DT_FLOAT, TensorShape({})); in Parse() 65 out_tensors->emplace_back(cpu_allocator(), DT_DOUBLE, TensorShape({})); in Parse() 70 out_tensors->emplace_back(cpu_allocator(), DT_BOOL, TensorShape({})); in Parse() 75 out_tensors->emplace_back(cpu_allocator(), DT_STRING, TensorShape({})); in Parse() 80 out_tensors->emplace_back(cpu_allocator(), DT_INT64, TensorShape({})); in Parse() [all …]
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | copy_tensor.cc | 50 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice() argument 55 Tensor copy(cpu_allocator, DT_VARIANT, input->shape()); in CopyHostToDevice() 64 [dst, recv_dev_context, out_allocator, status_cb, cpu_allocator, in CopyHostToDevice() 70 CopyHostToDevice(&from, cpu_allocator, out_allocator, edge_name, in CopyHostToDevice() 115 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost() argument 120 Tensor copy(cpu_allocator, DT_VARIANT, input->shape()); in CopyDeviceToHost() 130 cpu_allocator](StatusCallback wrapped_done_, in CopyDeviceToHost() 135 CopyDeviceToHost(&from, cpu_allocator, out_allocator, edge_name, in CopyDeviceToHost() 181 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice() argument 189 Tensor copy(cpu_allocator, DT_VARIANT, input->shape()); in CopyDeviceToDevice() [all …]
|
D | mkl_threadpool_device_test.cc | 34 options, "/device:CPU:0", Bytes(256), DeviceLocality(), cpu_allocator()); in TEST() 45 options, "/device:CPU:0", Bytes(256), DeviceLocality(), cpu_allocator()); in TEST()
|
D | single_threaded_cpu_device.cc | 69 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto() 77 return cpu_allocator(); in GetAllocator()
|
D | mkl_cpu_allocator.h | 279 return cpu_allocator()->AllocateRaw(kAlignment, size); in MallocHook() 284 cpu_allocator()->DeallocateRaw(ptr); in FreeHook()
|
/external/tensorflow/tensorflow/lite/delegates/flex/ |
D | buffer_map.cc | 36 proto->set_allocator_name(tensorflow::cpu_allocator()->Name()); in FillAllocationDescription() 48 data(), tensorflow::cpu_allocator()); in LogAllocation() 56 tensorflow::cpu_allocator(), false); in LogDeallocation() 66 : BaseTfLiteTensorBuffer(tensorflow::cpu_allocator()->AllocateRaw( in TfLiteTensorBuffer() 82 tensorflow::cpu_allocator()->DeallocateRaw(data()); in ~TfLiteTensorBuffer() 102 tensorflow::cpu_allocator()->Deallocate<string>( in ~StringTfLiteTensorBuffer() 112 ? tensorflow::cpu_allocator()->Allocate<string>(num_strings) in StringTfLiteTensorBuffer()
|
/external/tensorflow/tensorflow/core/framework/ |
D | allocator_test.cc | 90 Allocator* a = cpu_allocator(); in TEST() 129 Allocator* a = cpu_allocator(); in TEST() 139 Allocator* a = cpu_allocator(); in TEST() 150 Allocator* a = cpu_allocator(); in TEST() 176 Allocator* a = cpu_allocator(); in BM_Allocation()
|
D | allocator.cc | 192 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator() argument 193 : SubAllocator({}, {}), cpu_allocator_(cpu_allocator) {} in CPUSubAllocator() 224 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() function
|
D | op_kernel_test.cc | 161 std::move(device_type), &device_, cpu_allocator(), in ExpectSuccess() 177 CreateOpKernel(std::move(device_type), &device_, cpu_allocator(), in ExpectFailure() 314 return cpu_allocator(); in GetAllocator() 328 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F() 353 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F() 381 CreateOpKernel(DEVICE_CPU, params.device, cpu_allocator(), in TEST_F() 438 cpu_allocator(), def, 472 cpu_allocator(), def, in ExpectFailure() 664 DEVICE_CPU, params.device, cpu_allocator(), in TEST_F() 939 cpu_allocator(), node_def, in BM_InputRangeHelper()
|
D | op_segment_test.cc | 67 auto created = CreateOpKernel(DEVICE_CPU, &device_, cpu_allocator(), in GetFn()
|
D | resource_op_kernel_test.cc | 42 return cpu_allocator(); in GetAllocator()
|
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_device_factory.cc | 33 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice() argument 35 physical_device_desc, gpu_allocator, cpu_allocator, in GPUDevice() 67 Allocator* cpu_allocator) override { in CreateGPUDevice() argument 70 gpu_allocator, cpu_allocator); in CreateGPUDevice()
|
D | gpu_device.h | 57 Allocator* gpu_allocator, Allocator* cpu_allocator, 309 Allocator* cpu_allocator) = 0;
|
/external/tensorflow/tensorflow/core/common_runtime/sycl/ |
D | sycl_device.h | 117 m_cpu_allocator_.push_back(cpu_allocator()); in AddDevice() 201 Allocator* cpu_allocator, SYCLDeviceContext* ctx) in SYCLDevice() argument 205 cpu_allocator_(cpu_allocator), in SYCLDevice()
|
/external/tensorflow/tensorflow/compiler/jit/kernels/ |
D | xla_ops.cc | 478 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute() local 482 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() 484 Tensor compilation_successful(cpu_allocator, DT_BOOL, TensorShape({})); in Compute() 486 ctx->set_output(0, Tensor(cpu_allocator, DT_STRING, TensorShape({}))); in Compute() 499 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() 502 Tensor compilation_successful(cpu_allocator, DT_BOOL, TensorShape({})); in Compute()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | immutable_constant_op_test.cc | 44 : memptr_(cpu_allocator()->AllocateRaw(kTestAlignment, length)), in TestReadOnlyMemoryRegion() 47 cpu_allocator()->DeallocateRaw(memptr_); in ~TestReadOnlyMemoryRegion()
|
D | stack.cc | 255 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync() local 257 new Tensor(cpu_allocator, tensor.dtype(), tensor.shape()); in ComputeAsync()
|
D | constant_op_test.cc | 61 cpu_allocator(), const_node, in PersistentMemoryTrackingTest()
|
D | restore_op_test.cc | 89 cpu_allocator(), save, in TEST_F() 393 cpu_allocator(), save, in TEST_F()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | evaluation_utils.h | 45 return cpu_allocator(); in GetAllocator()
|
D | evaluation_utils.cc | 75 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto()
|
/external/tensorflow/tensorflow/c/kernels/ |
D | bitcast_op_test.cc | 30 return cpu_allocator(); in GetAllocator()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | const_op.cc | 120 OP_REQUIRES(ctx, tensor.FromProto(cpu_allocator(), proto_), in Compile()
|
/external/tensorflow/tensorflow/contrib/util/ |
D | convert_graphdef_memmapped_format_lib.cc | 69 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in ConvertConstantsToImmutable()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_device.cc | 247 return cpu_allocator(); in GetAllocatorLocked() 446 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto()
|