Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dpin_to_host_optimizer.cc169 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeOutputPortHostFriendly() local
206 for (const string& host_memory_arg : kernel->host_memory_arg()) { in IsNodeInputPortHostFriendly() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dkernel_def.proto32 repeated string host_memory_arg = 4; field
H A Dop_kernel.cc1766 for (const auto& host_memory_arg : kernel_def.host_memory_arg()) { in ValidateKernelRegistrations() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dkernel_def.pb.h884 inline const std::string& KernelDef::host_memory_arg(int index) const { in host_memory_arg() function
931 KernelDef::host_memory_arg() const { in host_memory_arg() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dexecute.cc894 for (const string& host_memory_arg : kernel_def->host_memory_arg()) { in IntArgsAndRetvalsOnDevice() local