Home
last modified time | relevance | path

Searched refs:device_attrs (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_multi_client_test.cc126 std::vector<tensorflow::DeviceAttributes> device_attrs; in TEST() local
129 context->ListDevices(&device_attrs); in TEST()
131 for (const auto& device_attr : device_attrs) { in TEST()
172 std::vector<tensorflow::DeviceAttributes> device_attrs; in TEST() local
175 context->ListDevices(&device_attrs); in TEST()
178 for (const auto& device_attr : device_attrs) { in TEST()
Dc_api_multi_client_function_test.cc240 std::vector<tensorflow::DeviceAttributes> device_attrs; in TEST_P() local
241 tensorflow::unwrap(ctx)->ListDevices(&device_attrs); in TEST_P()
243 for (const auto& device_attr : device_attrs) { in TEST_P()
/external/tensorflow/tensorflow/python/framework/
Dgpu_util.py31 def compute_capability_from_device_desc(device_attrs): argument
48 device_attrs.physical_device_desc)
/external/tensorflow/tensorflow/core/tfrt/utils/
Dutils.cc107 tensorflow::DeviceAttributes device_attrs = in CreateDummyTfDevices() local
112 tensorflow::Env::Default(), device_attrs)); in CreateDummyTfDevices()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc411 llvm::SmallVector<NamedAttribute, 8> device_attrs; in AssignDevicesToReplicate() local
418 device_attrs.push_back( in AssignDevicesToReplicate()
430 device_attrs.push_back(builder->getNamedAttr( in AssignDevicesToReplicate()
433 replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); in AssignDevicesToReplicate()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc1412 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attrs, in FilterDevicesForRemoteWorkers() argument
1414 filtered_device_mask->resize(device_attrs.size()); in FilterDevicesForRemoteWorkers()
1428 for (int i = 0; i < device_attrs.size(); i++) { in FilterDevicesForRemoteWorkers()
1430 DeviceNameUtils::ParseFullName(device_attrs[i].name(), &pn); in FilterDevicesForRemoteWorkers()
Dcontext.h478 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attrs,
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_600.h87623 struct transport_container device_attrs; member
Dvmlinux.h87623 struct transport_container device_attrs; member
/external/bcc/libbpf-tools/loongarch/
Dvmlinux_602.h43689 struct transport_container device_attrs; member
Dvmlinux.h43689 struct transport_container device_attrs; member
/external/bcc/libbpf-tools/x86/
Dvmlinux.h101572 struct transport_container device_attrs; member
Dvmlinux_518.h101572 struct transport_container device_attrs; member
/external/bcc/libbpf-tools/riscv/
Dvmlinux_602.h88932 struct transport_container device_attrs; member
Dvmlinux.h88932 struct transport_container device_attrs; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux_608.h98666 struct transport_container device_attrs; member
Dvmlinux.h98666 struct transport_container device_attrs; member