Home
last modified time | relevance | path

Searched defs:available_devices (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_test.cc73 std::vector<DeviceAttributes> available_devices; in RunAndFetch() local
Dfused_batch_norm_ex_op_test.cc112 std::vector<DeviceAttributes> available_devices; local
Dconv_ops_test.cc527 std::vector<DeviceAttributes> available_devices; in RunAndFetch() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1213 const absl::flat_hash_set<int64_t>& available_devices, in DevicesForShardingInternal()
1243 const HloSharding& sharding, absl::Span<const int64_t> available_devices) { in DevicesForSharding()