Searched refs:host_memory_arg (Results 1 – 4 of 4) sorted by relevance
166 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeOutputPortHostFriendly() local167 if (op->output_arg(output_arg_id).name() == host_memory_arg) { in IsNodeOutputPortHostFriendly()203 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeInputPortHostFriendly() local204 if (op->input_arg(input_arg_id).name() == host_memory_arg) { in IsNodeInputPortHostFriendly()
30 repeated string host_memory_arg = 4; field
118 const auto& from_proto = kdef->host_memory_arg(); in MemoryTypesForNode()
1360 for (const auto& host_memory_arg : kernel_def.host_memory_arg()) { in ValidateKernelRegistrations() local1361 if (!FindArgInOp(host_memory_arg, op_def.input_arg()) && in ValidateKernelRegistrations()1362 !FindArgInOp(host_memory_arg, op_def.output_arg())) { in ValidateKernelRegistrations()1364 "HostMemory arg '", host_memory_arg, in ValidateKernelRegistrations()