Searched refs:device_ordinals (Results 1 – 6 of 6) sorted by relevance
5 name: "device_ordinals"
219 std::vector<int> device_ordinals; in DeviceOrdinals() local220 device_ordinals.reserve(num_replicas); in DeviceOrdinals()222 device_ordinals.push_back(device_assignment_(i, 0)); in DeviceOrdinals()224 return device_ordinals; in DeviceOrdinals()
17799 func TPUOrdinalSelector(scope *Scope) (device_ordinals tf.Output) {
39849 name: "device_ordinals"
82694 name: "device_ordinals"
82067 name: "device_ordinals"