Searched defs:fast_mem (Results 1 – 6 of 6) sorted by relevance
70 std::optional<xla::OpSharding> sharding, bool fast_mem) { in ReshapeWithCorrectRepresentationAndSharding()
85 std::optional<xla::OpSharding> sharding, bool fast_mem) { in ReshapeWithCorrectRepresentationAndSharding()
98 bool fast_mem = false; member
57 bool fast_mem = 7; field
538 mlir::XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in ConvertMLIRToXlaComputation()
116 bool fast_mem; member1375 int device_ordinal, bool fast_mem) { in InitializeVarOnTPU()