Searched defs:layout_preference (Results 1 – 9 of 9) sorted by relevance
69 auto layout_preference = shape_determination_fns.layout_preference_fn( in RewriteLayoutWithShardedShape() local104 auto layout_preference = in ReshapeWithCorrectRepresentationAndSharding() local
310 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()356 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()411 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()1172 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()1215 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()1351 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()1424 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in TEST_F()
571 XlaLayoutPreference layout_preference = in ReadVariableInputTensor() local719 XlaLayoutPreference layout_preference = in AssignVariableTensor() local
139 XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in IdentityShapeRepresentationFn()
866 auto layout_preference = in XLAShapeForArgument() local899 auto layout_preference = in XLAShapeForArgument() local
58 XlaLayoutPreference layout_preference) { in TpuShapeRepresentation()205 XlaLayoutPreference layout_preference = in TpuDeviceToDeviceCopy() local
136 XlaLayoutPreference layout_preference = in CopyCPUTensorToDevice() local
139 auto layout_preference = shape_determination_fns.layout_preference_fn( in GetXlaInputShapes() local197 auto layout_preference = shape_determination_fns.layout_preference_fn( in GetOutputInfo() local538 mlir::XlaLayoutPreference layout_preference) -> StatusOr<xla::Shape> { in ConvertMLIRToXlaComputation()
468 auto layout_preference = shape_determination_fns.layout_preference_fn( in GetShardingInfo() local