Home
last modified time | relevance | path

Searched refs:devices_size (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dservice_test.cc39 EXPECT_EQ(global.nodes()[0].devices_size(), 2); in TEST()
40 EXPECT_EQ(global.nodes()[1].devices_size(), 2); in TEST()
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_trace_events_test.cc69 ASSERT_EQ(trace.devices_size(), 2); in TEST()
/external/OpenCL-CTS/test_common/harness/
DkernelHelpers.cpp1451 size_t devices_size = 0; in get_min_alignment() local
1457 &devices_size); in get_min_alignment()
1460 devices = (cl_device_id *)malloc(devices_size); in get_min_alignment()
1467 error = clGetContextInfo(context, CL_CONTEXT_DEVICES, devices_size, in get_min_alignment()
1471 for (i = 0; i < (int)(devices_size / sizeof(cl_device_id)); i++) in get_min_alignment()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc257 local_topology.devices_size()); in BuildDistributedDevices()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show.cc203 if (node->proto().devices_size() > 0) { in FormatNode()