Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable_run_options.h90 const DeviceAssignment* device_assn; // never null member
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplicated_io_feed_test.cc62 auto device_assn = MakeDeviceAssn(kNumReplicas); in XLA_TEST_F() local
Dcollective_ops_test.cc336 DeviceAssignment device_assn = MakeDeviceAssn(devices); in XLA_TEST_F() local
377 auto device_assn = MakeDeviceAssn(devices); in XLA_TEST_F() local