Searched defs:use_fast_memory (Results 1 – 4 of 4) sorted by relevance
136 bool use_fast_memory) -> StatusOr<xla::Shape> { in IdentityShapeRepresentationFn()145 const absl::optional<xla::HloSharding>& sharding, bool use_fast_memory, in RewriteLayoutWithShardedShape()
309 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()352 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()405 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()1167 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()1208 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()1342 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()1413 bool use_fast_memory) -> StatusOr<xla::Shape> { in TEST_F()
105 bool use_fast_memory) -> StatusOr<xla::Shape> { in XlaDeviceContext()
55 bool use_fast_memory) { in TpuShapeRepresentation()