Home
last modified time | relevance | path

Searched refs:set_gpu_compatible (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc55 aa.set_gpu_compatible(gpu_compatible); in TEST()
70 b.set_gpu_compatible(true); in TEST()
82 a.set_gpu_compatible(true); in TEST()
Dallocator.h364 void set_gpu_compatible(bool v) { value |= (static_cast<int>(v) << 2); } in set_gpu_compatible() function
/external/tensorflow/tensorflow/contrib/framework/kernels/
Dzero_initializer_op.cc47 attr.set_gpu_compatible(true); in Compute()
106 attr.set_gpu_compatible(true); in Compute()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc38 alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h58 attr.set_gpu_compatible(true); in EnsureSparseVariableAccess()
219 attr.set_gpu_compatible(true); in PrepareToUpdateVariable()
Dgpu_device_array.h58 attr.set_gpu_compatible(true); in Init()
Dassign_op.h59 attr.set_gpu_compatible(true); in Compute()
Dhistogram_op_gpu.cu.cc46 pinned_allocator.set_gpu_compatible(true); in Compute()
Dcount_up_to_op.cc85 attr.set_gpu_compatible(true); in Compute()
Ddense_update_functor.cc89 attr.set_gpu_compatible(true); \
Dcheck_numerics_op.cc160 attr.set_gpu_compatible(true); in ComputeAsync()
Dstack.cc253 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
Dcuda_solvers.h351 alloc_attr.set_gpu_compatible(true);
Ddynamic_partition_op_gpu.cu.cc283 alloc_attr.set_gpu_compatible(true); in ComputeAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc130 cpu_attr.set_gpu_compatible(true); in RecvFromPeer()
Dbase_rendezvous_mgr.cc269 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc101 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
Dcopy_tensor.cc268 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc119 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()
Dgdr_memory_manager.cc376 alloc_attrs.set_gpu_compatible(true); in TransportOptionsFromTensor()
459 alloc_attrs.set_gpu_compatible(true); in TensorFromTransportOptions()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc536 attr.set_gpu_compatible(true); in EnsureOutputAllocated()
581 attr.set_gpu_compatible(true); in ProcessResult()
694 attr.set_gpu_compatible(true); in ReadBatchResult()
Dnuma_map_and_batch_dataset_op.cc569 attr.set_gpu_compatible(true); in GetBatch()
957 attr.set_gpu_compatible(true); in AllocateOutput()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc497 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()
591 cpu_attr.set_gpu_compatible(true); in RecvBufAsync()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc269 allocator_attributes.set_gpu_compatible(true); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc476 host_alloc_attrs.set_gpu_compatible(true); in Compute()

12