Searched defs:on_host (Results 1 – 11 of 11) sorted by relevance
91 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local
367 Device* device, DeviceContext* device_context, bool on_host, in TransportOptionsFromTensor()425 Device* device, DeviceContext* device_context, bool on_host, in TensorFromTransportOptions()
63 const bool on_host = recv_args_.alloc_attrs.on_host(); in Start() local
57 const bool on_host = in SetOutputAttrs() local
491 const bool on_host = send_args.alloc_attrs.on_host(); in GrpcRecvTensorAsync() local583 const bool on_host = in RecvBufAsync() local
129 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in Initialize() local
201 const bool on_host = in CreateOpKernelContext() local
361 bool on_host() const { return value & 0x1; } in on_host() function
1060 const bool on_host = send_args.alloc_attrs.on_host(); in RecvHandler() local1566 bool on_host = recv_args_.alloc_attrs.on_host(); in AllocateTensorsAsync() local
1076 const bool on_host = IsDstInputOnHost(edge, g_info); in Partition() local
770 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY; in SetAllocAttrs() local