Searched refs:lu_shape (Results 1 – 3 of 3) sorted by relevance
43 Shape lu_shape = ShapeUtil::MakeTupleShape( in LuDecomposition() local50 return CustomCall(a.builder(), "LuDecomposition", {a}, lu_shape); in LuDecomposition()
72 lu_shape = np.array(lu.shape.as_list())73 batch_shape = lu_shape[:-2]74 num_rows = lu_shape[-2]75 num_cols = lu_shape[-1]121 lu_shape)
196 ShapeHandle lu_shape; in LuShapeFn() local199 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(n, n), &lu_shape)); in LuShapeFn()202 c->set_output(0, lu_shape); in LuShapeFn()