Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/proto/
Dlayout.proto50 repeated int64 global_device_ids = 2; field
/external/tensorflow/tensorflow/dtensor/cc/
Dtensor_layout.h168 absl::Span<const int64_t> global_device_ids() const { in global_device_ids() function
Dtensor_layout.cc253 const std::vector<std::int64_t>& global_device_ids, in GetMesh()
535 std::vector<int64_t> global_device_ids; in GenerateMeshDevicesForTests() local
582 std::vector<int64_t> global_device_ids; in FromString() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc226 std::vector<xla::GlobalDeviceId> global_device_ids( in ResolveDeviceAssignment() local