Searched defs:on_host (Results 1 – 11 of 11) sorted by relevance
53 TF_Bool on_host; member
202 bool on_host = in Initialize() local229 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
449 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs() local
61 const bool on_host = in SetOutputAttrs() local
372 bool on_host = DataTypeAlwaysOnHost(dtype); in ComputeAsync() local
530 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local667 const bool on_host = in RecvBufAsync() local
315 bool on_host() const { return value & 0x1; } in on_host() function
398 TFE_TensorHandle* on_host = in TEST() local
269 const bool on_host = in SetOutputAttrs() local
1104 const bool on_host = IsDstInputOnHost(edge, g_info); in Partition() local
519 const bool on_host = in CreateOpKernelContext() local