Home
last modified time | relevance | path

Searched defs:host_memory_arg (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dkernel_def.proto30 repeated string host_memory_arg = 4; field
Dop_kernel.cc1360 for (const auto& host_memory_arg : kernel_def.host_memory_arg()) { in ValidateKernelRegistrations() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer.cc166 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeOutputPortHostFriendly() local
203 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeInputPortHostFriendly() local