Searched refs:input_devices_ (Results 1 – 3 of 3) sorted by relevance
116 input_devices_.resize(kernel_->num_inputs(), device_); in Init()121 input_devices_[i] = host_cpu_device_; in Init()160 for (const Device* device : input_devices_) { in InstantiateFunc()430 return input_devices_[i]; in InputDevice()435 (composite_devices_.find(input_devices_[i]->name()) == in InputDevice()439 return input_devices_[i]; in InputDevice()
245 std::vector<Device*> input_devices_; variable277 input_devices_(std::move(input_devices)), in KernelAndDeviceFunc()340 std::vector<Device*> input_devices_; variable
154 : input_devices_(InputDevices(caller)) {} in DefaultFunctionBodyPlacer()157 return input_devices_[input_index]; in InputNodeDevice()171 const std::vector<string> input_devices_; member in tensorflow::__anon12cc2c770111::DefaultFunctionBodyPlacer206 input_devices_(InputDevices(caller)) { in MultiDeviceFunctionBodyPlacer()212 return input_devices_[input_index]; in InputNodeDevice()257 std::vector<string> input_devices_; member in tensorflow::__anon12cc2c770111::MultiDeviceFunctionBodyPlacer