Home
last modified time | relevance | path

Searched defs:host_memory_args (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmemory_types.cc46 std::vector<string>* host_memory_args, in MemoryTypesHelper()
128 std::vector<string> host_memory_args(from_proto.begin(), from_proto.end()); in MemoryTypesForNode() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dexecute.cc313 const auto& host_memory_args = kernel_def->host_memory_arg(); in IsHostMemoryArg() local