Searched defs:use_fast_memory (Results 1 – 4 of 4) sorted by relevance
131 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in IdentityShapeRepresentationFn()140 const absl::optional<xla::HloSharding>& sharding, bool use_fast_memory, in RewriteLayoutWithShardedShape()
308 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()351 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()404 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()1100 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()1141 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()1275 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()1346 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in TEST_F()
98 bool use_fast_memory) -> xla::StatusOr<xla::Shape> { in XlaDeviceContext()
55 bool use_fast_memory) { in TpuShapeRepresentation()