Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/util/
Dtensor_ops_util.h37 attr.set_on_host(true); in ZerosLikeTensor()
98 attr.set_on_host(true); in BinaryAddTensors()
Drocm_solvers.h116 alloc_attr.set_on_host(true);
/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()
441 attr.set_on_host(true); in Compute()
480 attr.set_on_host(true); in Compute()
620 attr.set_on_host(true); in Compute()
655 attr.set_on_host(true); in Compute()
775 attr.set_on_host(true); in Compute()
977 attr.set_on_host(true); in Compute()
Dhost_constant_op.cc31 alloc_attr.set_on_host(true); in _HostConstantOp()
Dgpu_device_array.h59 attr.set_on_host(true); in Init()
Dmap_kernels.h70 attr.set_on_host(true); in ForwardInputOrCreateNewMap()
86 attr.set_on_host(true); in Compute()
Dtraining_op_helpers.h48 attr.set_on_host(true); in EnsureSparseVariableAccess()
207 attr.set_on_host(true); in PrepareToUpdateVariable()
Dhistogram_op_gpu.cu.cc45 pinned_allocator.set_on_host(true); in Compute()
Dstack.cc196 alloc_attr.set_on_host(true); in Compute()
255 host_alloc_attrs.set_on_host(true); in ComputeAsync()
Dfunction_ops.cc339 arg_alloc_attrs.set_on_host(DataTypeAlwaysOnHost(dtype)); in ComputeAsync()
345 ret_alloc_attrs.set_on_host(DataTypeAlwaysOnHost(dtype)); in ComputeAsync()
Dresource_variable_ops.cc99 attr.set_on_host(true); in CopyVariable()
230 attr.set_on_host(true); in VarHandleOp()
242 attr.set_on_host(true); in Compute()
476 attr.set_on_host(true); in Compute()
Ddynamic_partition_op_gpu.cu.cc253 alloc_attr.set_on_host(true); in ComputeAsync()
289 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()
Dsingle_threaded_executor.cc190 kernel_state.output_alloc_attr.set_on_host(on_host); in Initialize()
218 h.set_on_host(on_host); in Initialize()
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc135 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
152 alloc_attr.set_on_host(true); in UpdateArgAndRetvalMetadata()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dzeros_op.cc55 cpu_alloc.set_on_host(true); in Compute()
Dsparse_tensor_to_csr_sparse_matrix_op.cc104 cpu_alloc.set_on_host(true); in Compute()
271 cpu_alloc.set_on_host(true); in ComputeAsync()
Ddense_to_csr_sparse_matrix_op.cc125 cpu_alloc.set_on_host(true); in Compute()
338 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/kernels/batching_util/
Dconcat_split_util.h75 attr.set_on_host(true); in Concat()
173 attr.set_on_host(true); in SplitCPU()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc78 host_alloc_attrs.set_on_host(true); in RunComposedOp()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc119 input_alloc_attrs_[i].set_on_host(host); in Init()
126 output_alloc_attrs_[i].set_on_host(kernel_->output_memory_types()[i] == in Init()
/external/tensorflow/tensorflow/core/framework/
Dallocator_test.cc54 aa.set_on_host(on_host); in TEST()

123