HomeSort by relevance Sort by last modified time
    Searched defs:localities (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
device_resolver_local_test.cc 50 std::vector<DeviceLocality> localities; local
57 drl_->GetDeviceLocalitiesAsync(cp.instance, &localities,
64 EXPECT_EQ(2, localities.size());
69 std::vector<DeviceLocality> localities; local
75 drl_->GetDeviceLocalitiesAsync(cp.instance, &localities,
82 EXPECT_EQ(0, localities.size());
collective_param_resolver_local.cc 158 // Create a populated GlobalDeviceMap from CollInstanceParams and localities.
160 const std::vector<DeviceLocality>& localities) {
163 CHECK_EQ(ip.device_names.size(), localities.size());
172 dr->locality = &localities[i];
293 CollectiveParams* cp, const std::vector<DeviceLocality>& localities) {
295 GlobalDeviceMap gdm = BuildDevRecs(cp->instance, localities);
433 std::vector<DeviceLocality>* localities = new std::vector<DeviceLocality>; local
435 ir->shared.instance, localities,
436 [this, gr, cp, ir, localities, done](const Status& s)
446 CompleteDefaultRanking(gr, cp, ir, *localities);
    [all...]

Completed in 182 milliseconds