Searched defs:on_host (Results 1 – 10 of 10) sorted by relevance
53 TF_Bool on_host; member
188 bool on_host = in Initialize() local215 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
514 const bool on_host = in CreateOpKernelContext() local
61 const bool on_host = in SetOutputAttrs() local
525 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local662 const bool on_host = in RecvBufAsync() local
432 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs() local
367 bool on_host() const { return value & 0x1; } in on_host() function
395 TFE_TensorHandle* on_host = in TEST() local
269 const bool on_host = in SetOutputAttrs() local
1103 const bool on_host = IsDstInputOnHost(edge, g_info); in Partition() local