Searched refs:host_memory_arg (Results 1 – 5 of 5) sorted by relevance
169 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeOutputPortHostFriendly() local170 if (op->output_arg(output_arg_id).name() == host_memory_arg) { in IsNodeOutputPortHostFriendly()206 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeInputPortHostFriendly() local207 if (op->input_arg(input_arg_id).name() == host_memory_arg) { in IsNodeInputPortHostFriendly()
32 repeated string host_memory_arg = 4; field
127 const auto& from_proto = kdef->host_memory_arg(); in MemoryTypesForNode()
1766 for (const auto& host_memory_arg : kernel_def.host_memory_arg()) { in ValidateKernelRegistrations() local1767 if (!FindArgInOp(host_memory_arg, op_def.input_arg()) && in ValidateKernelRegistrations()1768 !FindArgInOp(host_memory_arg, op_def.output_arg())) { in ValidateKernelRegistrations()1770 "HostMemory arg '", host_memory_arg, in ValidateKernelRegistrations()
313 const auto& host_memory_args = kernel_def->host_memory_arg(); in IsHostMemoryArg()894 for (const string& host_memory_arg : kernel_def->host_memory_arg()) { in IntArgsAndRetvalsOnDevice() local896 if (output_arg.name() == host_memory_arg) { in IntArgsAndRetvalsOnDevice()