UNPKG

apache-arrow

Version:
37 lines (35 loc) 2.21 kB
"use strict"; // automatically generated by the FlatBuffers compiler, do not modify Object.defineProperty(exports, "__esModule", { value: true }); exports.unionListToSparseTensorIndex = exports.unionToSparseTensorIndex = exports.SparseTensorIndex = void 0; const sparse_matrix_index_c_s_x_js_1 = require("./sparse-matrix-index-c-s-x.js"); const sparse_tensor_index_c_o_o_js_1 = require("./sparse-tensor-index-c-o-o.js"); const sparse_tensor_index_c_s_f_js_1 = require("./sparse-tensor-index-c-s-f.js"); var SparseTensorIndex; (function (SparseTensorIndex) { SparseTensorIndex[SparseTensorIndex["NONE"] = 0] = "NONE"; SparseTensorIndex[SparseTensorIndex["SparseTensorIndexCOO"] = 1] = "SparseTensorIndexCOO"; SparseTensorIndex[SparseTensorIndex["SparseMatrixIndexCSX"] = 2] = "SparseMatrixIndexCSX"; SparseTensorIndex[SparseTensorIndex["SparseTensorIndexCSF"] = 3] = "SparseTensorIndexCSF"; })(SparseTensorIndex = exports.SparseTensorIndex || (exports.SparseTensorIndex = {})); function unionToSparseTensorIndex(type, accessor) { switch (SparseTensorIndex[type]) { case 'NONE': return null; case 'SparseTensorIndexCOO': return accessor(new sparse_tensor_index_c_o_o_js_1.SparseTensorIndexCOO()); case 'SparseMatrixIndexCSX': return accessor(new sparse_matrix_index_c_s_x_js_1.SparseMatrixIndexCSX()); case 'SparseTensorIndexCSF': return accessor(new sparse_tensor_index_c_s_f_js_1.SparseTensorIndexCSF()); default: return null; } } exports.unionToSparseTensorIndex = unionToSparseTensorIndex; function unionListToSparseTensorIndex(type, accessor, index) { switch (SparseTensorIndex[type]) { case 'NONE': return null; case 'SparseTensorIndexCOO': return accessor(index, new sparse_tensor_index_c_o_o_js_1.SparseTensorIndexCOO()); case 'SparseMatrixIndexCSX': return accessor(index, new sparse_matrix_index_c_s_x_js_1.SparseMatrixIndexCSX()); case 'SparseTensorIndexCSF': return accessor(index, new sparse_tensor_index_c_s_f_js_1.SparseTensorIndexCSF()); default: return null; } } exports.unionListToSparseTensorIndex = unionListToSparseTensorIndex; //# sourceMappingURL=sparse-tensor-index.js.map