Home
last modified time | relevance | path

Searched refs:set_on_host (Results 1 – 25 of 48) 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.cc110 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
125 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc120 attr.set_on_host(true); in ForwardInputOrCreateNewList()
144 attr.set_on_host(true); in Compute()
311 attr.set_on_host(true); in Compute()
361 attr.set_on_host(true); in Compute()
492 attr.set_on_host(true); in Compute()
Dlist_kernels.h133 attr.set_on_host(true); in Compute()
214 attr.set_on_host(true); in Compute()
261 attr.set_on_host(true); in Compute()
443 attr.set_on_host(true); in Compute()
482 attr.set_on_host(true); in Compute()
622 attr.set_on_host(true); in Compute()
657 attr.set_on_host(true); in Compute()
777 attr.set_on_host(true); in Compute()
979 attr.set_on_host(true); in Compute()
Dhost_constant_op.cc31 alloc_attr.set_on_host(true); in _HostConstantOp()
Dtraining_op_helpers.h48 attr.set_on_host(true); in EnsureSparseVariableAccess()
207 attr.set_on_host(true); in PrepareToUpdateVariable()
Dgpu_device_array.h58 attr.set_on_host(true); in Init()
Dhistogram_op_gpu.cu.cc55 pinned_allocator.set_on_host(true); in Compute()
Dstack.cc195 alloc_attr.set_on_host(true); in Compute()
254 host_alloc_attrs.set_on_host(true); in ComputeAsync()
Drocm_solvers.h115 alloc_attr.set_on_host(true);
Dfunction_ops.cc384 arg_alloc_attrs.set_on_host(true); in ComputeAsync()
391 ret_alloc_attrs.set_on_host(true); in ComputeAsync()
Dresource_variable_ops.cc99 attr.set_on_host(true); in CopyVariable()
237 attr.set_on_host(true); in VarHandleOp()
249 attr.set_on_host(true); in Compute()
491 attr.set_on_host(true); in Compute()
Ddynamic_partition_op_gpu.cu.cc257 alloc_attr.set_on_host(true); in ComputeAsync()
293 alloc_attr.set_on_host(true); in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cc143 cpu_alloc.set_on_host(true); in WriteOptionalWithValueToOutput()
154 cpu_alloc.set_on_host(true); in WriteOptionalNoneToOutput()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dzeros_op.cc55 cpu_alloc.set_on_host(true); in Compute()
Dsparse_tensor_to_csr_sparse_matrix_op.cc103 cpu_alloc.set_on_host(true); in Compute()
270 cpu_alloc.set_on_host(true); in ComputeAsync()
Ddense_to_csr_sparse_matrix_op.cc124 cpu_alloc.set_on_host(true); in Compute()
337 cpu_alloc.set_on_host(true); in ComputeAsync()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc39 alloc_attrs.set_on_host(true); in ComputeAsync()
Devaluation_utils.cc86 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/compiler/tests/
Dunary_ops_composition_test.cc72 host_alloc_attrs.set_on_host(true); in RunComposedOp()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc106 input_alloc_attrs_[i].set_on_host(kernel_->input_memory_types()[i] == in Init()
111 output_alloc_attrs_[i].set_on_host(kernel_->output_memory_types()[i] == in Init()
/external/tensorflow/tensorflow/core/framework/
Dallocator.h307 void set_on_host(bool v) { value |= (static_cast<int>(v)); } in set_on_host() function
Dresource_mgr.h819 attr.set_on_host(true);
832 attr.set_on_host(true);
868 attr.set_on_host(true);
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops.cc55 attr.set_on_host(true); in Compute()

12