HomeSort by relevance Sort by last modified time
    Searched full:set_on_host (Results 1 - 25 of 26) sorted by null

1 2

  /external/tensorflow/tensorflow/core/kernels/
list_kernels.cc 149 attr.set_on_host(true);
214 attr.set_on_host(true);
330 attr.set_on_host(true);
365 attr.set_on_host(true);
457 attr.set_on_host(true);
cuda_device_array.h 57 attr.set_on_host(true);
histogram_op_gpu.cu.cc 45 pinned_allocator.set_on_host(true);
dynamic_partition_op_gpu.cu.cc 245 alloc_attr.set_on_host(true);
281 alloc_attr.set_on_host(true);
stack_ops.cc 205 alloc_attr.set_on_host(true);
293 host_alloc_attrs.set_on_host(true);
check_numerics_op.cc 147 attr.set_on_host(true);
list_kernels.h 141 attr.set_on_host(true);
ops_testutil.h 59 attr.set_on_host(on_host);
constant_op.cc 136 alloc_attr.set_on_host(true);
crop_and_resize_op.cc 749 alloc_attr.set_on_host(true);
cuda_solvers.h 341 alloc_attr.set_on_host(true);
tensor_array_ops.cc 111 alloc_attr.set_on_host(true);
    [all...]
  /external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_device.cc 50 attr.set_on_host(true);
  /external/tensorflow/tensorflow/contrib/gdr/
gdr_worker.cc 124 alloc_attrs.set_on_host(true);
  /external/tensorflow/tensorflow/core/framework/
allocator_test.cc 50 aa.set_on_host(on_host);
allocator.h 353 // attr.set_on_host(true);
356 void set_on_host(bool v) { value |= (static_cast<int>(v)); } function in struct:tensorflow::AllocatorAttributes
  /external/tensorflow/tensorflow/c/eager/
runtime.cc 273 out_attrs[i].set_on_host(kernel_->output_memory_types()[i] ==
c_api.cc 220 attr.set_on_host(true);
    [all...]
  /external/tensorflow/tensorflow/contrib/data/kernels/
prefetching_kernels.cc 178 arg_alloc_attr.set_on_host(true);
  /external/tensorflow/tensorflow/core/common_runtime/
copy_tensor.cc 254 host_alloc_attrs.set_on_host(true);
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_launch_op.cc 91 allocator_attrs.set_on_host(false);
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_worker_service.cc 415 alloc_attrs.set_on_host(true);
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 586 attr.set_on_host(true);
    [all...]
  /external/tensorflow/tensorflow/contrib/verbs/
rdma.cc     [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
constant_folding.cc 880 attr.set_on_host(true)
    [all...]

Completed in 1325 milliseconds

1 2