Searched refs:host_memory_arg (Results 1 – 4 of 4) sorted by relevance
170 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeOutputPortHostFriendly() local171 if (op->output_arg(output_arg_id).name() == host_memory_arg) { in IsNodeOutputPortHostFriendly()207 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeInputPortHostFriendly() local208 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()
1698 for (const auto& host_memory_arg : kernel_def.host_memory_arg()) { in ValidateKernelRegistrations() local1699 if (!FindArgInOp(host_memory_arg, op_def.input_arg()) && in ValidateKernelRegistrations()1700 !FindArgInOp(host_memory_arg, op_def.output_arg())) { in ValidateKernelRegistrations()1702 "HostMemory arg '", host_memory_arg, in ValidateKernelRegistrations()