Searched refs:host_memory_arg (Results 1 – 4 of 4) 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()
30 repeated string host_memory_arg = 4; field
141 const auto& from_proto = kdef->host_memory_arg(); in MemoryTypesForNode()
1648 for (const auto& host_memory_arg : kernel_def.host_memory_arg()) { in ValidateKernelRegistrations() local1649 if (!FindArgInOp(host_memory_arg, op_def.input_arg()) && in ValidateKernelRegistrations()1650 !FindArgInOp(host_memory_arg, op_def.output_arg())) { in ValidateKernelRegistrations()1652 "HostMemory arg '", host_memory_arg, in ValidateKernelRegistrations()