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.cc56 aa.set_gpu_compatible(gpu_compatible); in TEST()
71 b.set_gpu_compatible(true); in TEST()
83 a.set_gpu_compatible(true); in TEST()
Dallocator.h370 void set_gpu_compatible(bool v) { value |= (static_cast<int>(v) << 2); } in set_gpu_compatible() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc38 alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/
Dgpu_device_array.h60 attr.set_gpu_compatible(true); in Init()
Dtraining_op_helpers.h59 attr.set_gpu_compatible(true); in EnsureSparseVariableAccess()
218 attr.set_gpu_compatible(true); in PrepareToUpdateVariable()
Dhistogram_op_gpu.cu.cc46 pinned_allocator.set_gpu_compatible(true); in Compute()
Dassign_op.h59 attr.set_gpu_compatible(true); in Compute()
Dcount_up_to_op.cc84 attr.set_gpu_compatible(true); in Compute()
Ddense_update_functor.cc89 attr.set_gpu_compatible(true); \
Dstack.cc254 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
Dcheck_numerics_op.cc240 attr.set_gpu_compatible(true); in ComputeAsync()
Dinplace_ops.cc115 attr.set_gpu_compatible(true); in Compute()
Ddynamic_partition_op_gpu.cu.cc290 alloc_attr.set_gpu_compatible(true); in ComputeAsync()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc77 host_alloc_attrs.set_gpu_compatible(true); in RunComposedOp()
/external/tensorflow/tensorflow/core/util/
Drocm_solvers.h117 alloc_attr.set_gpu_compatible(true);
Dcuda_solvers.h394 alloc_attr.set_gpu_compatible(true);
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc361 attr->set_gpu_compatible(true); in InferAllocAttr()
388 attr->set_gpu_compatible(true); in InferAllocAttr()
Drendezvous_mgr.cc80 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
Dcopy_tensor.cc221 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc134 cpu_attr.set_gpu_compatible(true); in RecvFromPeer()
Dbase_rendezvous_mgr.cc265 attr.set_gpu_compatible(send_args.alloc_attrs.gpu_compatible() || in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc531 alloc_attrs.set_gpu_compatible(true); in GrpcRecvTensorAsync()
670 cpu_attr.set_gpu_compatible(true); in RecvBufAsync()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc734 attr.set_gpu_compatible(true); in ReadBatch()
830 attr.set_gpu_compatible(true); in ProcessBatch()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc396 allocator_attributes.set_gpu_compatible(true); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc422 host_alloc_attrs.set_gpu_compatible(true); in Compute()

12