Searched refs:UniqueDevice (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_test.cc55 EXPECT_IS_NOT_OK(sharding.UniqueDevice());
64 EXPECT_EQ(5, sharding.UniqueDevice().ValueOrDie());
136 EXPECT_IS_NOT_OK(sharding.UniqueDevice());
H A Dhlo_tfgraph_builder.cc108 "dev", instruction->sharding().UniqueDevice().ConsumeValueOrDie());
220 TF_ASSIGN_OR_RETURN(int64 device, instruction->sharding().UniqueDevice());
H A Dhlo_sharding.cc127 StatusOr<int64> HloSharding::UniqueDevice() const { function in class:HloSharding
131 "UniqueDevice() called on empty tuple");
136 [](const HloSharding& s) { return s.UniqueDevice(); });
152 "UniqueDevice() called on sharding that executes on multiple devices");
157 return UniqueDevice().status().ok();
H A Dhlo_sharding.h150 StatusOr<int64> UniqueDevice() const;
/external/vulkan-validation-layers/include/vulkan/
H A Dvulkan.hpp[all...]

Completed in 570 milliseconds