Home
last modified time | relevance | path

Searched refs:set_on_host (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/core/util/
Dtensor_ops_util.h37 attr.set_on_host(true); in ZerosLikeTensor()
98 attr.set_on_host(true); in BinaryAddTensors()
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc105 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
116 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h154 attr.set_on_host(true); in Compute()
235 attr.set_on_host(true); in Compute()
282 attr.set_on_host(true); in Compute()
464 attr.set_on_host(true); in Compute()
503 attr.set_on_host(true); in Compute()
643 attr.set_on_host(true); in Compute()
678 attr.set_on_host(true); in Compute()
798 attr.set_on_host(true); in Compute()
988 attr.set_on_host(true); in Compute()
Dlist_kernels.cc200 attr.set_on_host(true); in ForwardInputOrCreateNewList()
234 attr.set_on_host(true); in Compute()
394 attr.set_on_host(true); in Compute()
429 attr.set_on_host(true); in Compute()
555 attr.set_on_host(true); in Compute()
Dhost_constant_op.cc31 alloc_attr.set_on_host(true); in _HostConstantOp()
Dtraining_op_helpers.h47 attr.set_on_host(true); in EnsureSparseVariableAccess()
208 attr.set_on_host(true); in PrepareToUpdateVariable()
Dgpu_device_array.h57 attr.set_on_host(true); in Init()
Dstack.cc195 alloc_attr.set_on_host(true); in Compute()
254 host_alloc_attrs.set_on_host(true); in ComputeAsync()
Dhistogram_op_gpu.cu.cc45 pinned_allocator.set_on_host(true); in Compute()
Dfunction_ops.cc365 arg_alloc_attrs.set_on_host(true); in ComputeAsync()
372 ret_alloc_attrs.set_on_host(true); in ComputeAsync()
Dcheck_numerics_op.cc159 attr.set_on_host(true); in ComputeAsync()
Ddynamic_partition_op_gpu.cu.cc246 alloc_attr.set_on_host(true); in ComputeAsync()
282 alloc_attr.set_on_host(true); in ComputeAsync()
Dops_testutil.h59 attr.set_on_host(on_host); in SetOutputAttrs()
Dscale_and_translate_op.cc66 alloc_attr.set_on_host(true); in ComputeSpansCore()
158 alloc_attr.set_on_host(true); in ComputeGradSpansCore()
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc140 cpu_alloc.set_on_host(true); in WriteOptionalWithValueToOutput()
151 cpu_alloc.set_on_host(true); in WriteOptionalNoneToOutput()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc39 alloc_attrs.set_on_host(true); in ComputeAsync()
Devaluation_utils.cc107 attr.set_on_host(true); in EvaluateNode()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.cc49 attr.set_on_host(true); in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/framework/
Dresource_mgr.h645 attr.set_on_host(true);
658 attr.set_on_host(true);
694 attr.set_on_host(true);
Dallocator_test.cc53 aa.set_on_host(on_host); in TEST()
Dallocator.h360 void set_on_host(bool v) { value |= (static_cast<int>(v)); } in set_on_host() function
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc213 in_attrs[i].set_on_host(kernel_->input_memory_types()[i] == in Run()
218 out_attrs[i].set_on_host(kernel_->output_memory_types()[i] == in Run()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc120 alloc_attrs.set_on_host(true); in GrpcRecvTensorAsync()
Dgdr_memory_manager.cc378 alloc_attrs.set_on_host(true); in TransportOptionsFromTensor()
461 alloc_attrs.set_on_host(true); in TensorFromTransportOptions()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc141 attrs.set_on_host(true); in Compile()

12