Home
last modified time | relevance | path

Searched defs:on_host (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/c/
Dtf_tensor.h53 TF_Bool on_host; member
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc188 bool on_host = in Initialize() local
215 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
Ddataset_test_base.cc514 const bool on_host = in CreateOpKernelContext() local
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.cc61 const bool on_host = in SetOutputAttrs() local
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc525 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local
662 const bool on_host = in RecvBufAsync() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc432 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs() local
/external/tensorflow/tensorflow/core/framework/
Dallocator.h367 bool on_host() const { return value & 0x1; } in on_host() function
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc395 TFE_TensorHandle* on_host = in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc269 const bool on_host = in SetOutputAttrs() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc1103 const bool on_host = IsDstInputOnHost(edge, g_info); in Partition() local