- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for add_host_memory_arg (0.13 sec)
-
tensorflow/compiler/jit/xla_device.cc
const std::unordered_set<std::string>* constant_inputs = XlaOpRegistry::CompileTimeConstantInputArgNames(def->op()); for (const std::string& arg_name : *constant_inputs) { def->add_host_memory_arg(arg_name); } def->set_device_type(device); registrations->op_kernel_registrars.emplace_back( new kernel_factory::OpKernelRegistrar(def, kernel_class_name, factory)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0)