| /external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| D | c_api_multi_client_test.cc | 126 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()
|
| D | c_api_multi_client_function_test.cc | 240 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/ |
| D | gpu_util.py | 31 def compute_capability_from_device_desc(device_attrs): argument
48 device_attrs.physical_device_desc)
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | utils.cc | 107 tensorflow::DeviceAttributes device_attrs = in CreateDummyTfDevices() local
112 tensorflow::Env::Default(), device_attrs)); in CreateDummyTfDevices()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 411 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/ |
| D | context.cc | 1412 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()
|
| D | context.h | 478 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attrs,
|
| /external/bcc/libbpf-tools/powerpc/ |
| D | vmlinux_600.h | 87623 struct transport_container device_attrs; member
|
| D | vmlinux.h | 87623 struct transport_container device_attrs; member
|
| /external/bcc/libbpf-tools/loongarch/ |
| D | vmlinux_602.h | 43689 struct transport_container device_attrs; member
|
| D | vmlinux.h | 43689 struct transport_container device_attrs; member
|
| /external/bcc/libbpf-tools/x86/ |
| D | vmlinux.h | 101572 struct transport_container device_attrs; member
|
| D | vmlinux_518.h | 101572 struct transport_container device_attrs; member
|
| /external/bcc/libbpf-tools/riscv/ |
| D | vmlinux_602.h | 88932 struct transport_container device_attrs; member
|
| D | vmlinux.h | 88932 struct transport_container device_attrs; member
|
| /external/bcc/libbpf-tools/arm64/ |
| D | vmlinux_608.h | 98666 struct transport_container device_attrs; member
|
| D | vmlinux.h | 98666 struct transport_container device_attrs; member
|