Home
last modified time | relevance | path

Searched refs:index_dtype (Results 1 – 18 of 18) sorted by relevance

/dports/misc/py-gluonnlp/gluon-nlp-0.10.0/docs/examples/word_embedding/
H A Ddata.py135 index_dtype='int64'): argument
209 weight_dtype=dtype, index_dtype=index_dtype)
248 subword_lookup=subword_lookup, dtype=dtype, index_dtype=index_dtype)
306 weight_dtype=dtype, index_dtype=index_dtype)
315 dtype=dtype, index_dtype=index_dtype)
321 index_dtype): argument
325 centers = mx.nd.array(centers, dtype=index_dtype)
333 dtype, index_dtype): argument
337 centers = mx.nd.array(centers, dtype=index_dtype)
347 centers = mx.nd.array(centers, dtype=index_dtype)
[all …]
/dports/misc/py-gluonnlp/gluon-nlp-0.10.0/scripts/word_embeddings/
H A Ddata.py135 index_dtype='int64'): argument
209 weight_dtype=dtype, index_dtype=index_dtype)
248 subword_lookup=subword_lookup, dtype=dtype, index_dtype=index_dtype)
306 weight_dtype=dtype, index_dtype=index_dtype)
315 dtype=dtype, index_dtype=index_dtype)
321 index_dtype): argument
325 centers = mx.nd.array(centers, dtype=index_dtype)
333 dtype, index_dtype): argument
337 centers = mx.nd.array(centers, dtype=index_dtype)
347 centers = mx.nd.array(centers, dtype=index_dtype)
[all …]
H A Dtrain_glove.py287 index_dtype = 'int32'
289 index_dtype = 'int64'
292 indices = mx.nd.arange(counts.shape[0], dtype=index_dtype)
/dports/devel/py-pyopencl/pyopencl-2018.2/pyopencl/
H A Dalgorithm.py290 index_dtype.type.__name__, bits)
437 self.index_dtype = np.dtype(index_dtype)
847 def get_scan_kernel(self, index_dtype): argument
850 self.context, index_dtype,
858 def get_compress_kernel(self, index_dtype): argument
874 self.context, index_dtype,
895 def get_count_kernel(self, index_dtype): argument
952 def get_write_kernel(self, index_dtype): argument
1071 index_dtype = np.int64
1073 index_dtype = np.int32
[all …]
H A Dscan.py946 index_dtype=np.int32, argument
1043 self.index_dtype = np.dtype(index_dtype)
1044 if np.iinfo(self.index_dtype).min >= 0:
1092 index_dtype=self.index_dtype,
1104 index_typename = dtype_to_ctype(self.index_dtype)
1118 index_dtype,
1320 VectorArg(self.index_dtype,
1364 + [self.index_dtype, self.index_dtype, None, None])
1402 + self.index_dtype.itemsize*wg_size
1441 (None, self.index_dtype, self.index_dtype))
[all …]
/dports/misc/py-gluonnlp/gluon-nlp-0.10.0/tests/unittest/batchify/
H A Dtest_batchify_embedding.py57 def test_center_context_batchify(cbow, dtype, weight_dtype, index_dtype): argument
63 index_dtype=index_dtype)
70 assert contexts_row.dtype == index_dtype
89 def test_center_context_batchify_robustness(cbow, dtype, weight_dtype, index_dtype, emptyatbegin): argument
92 index_dtype=index_dtype)
104 assert contexts_row.dtype == index_dtype
/dports/www/grafana8/grafana-8.3.6/vendor/github.com/apache/arrow/python/pyarrow/tests/
H A Dtest_serialization.py550 index_dtype = np.dtype(index_type)
555 ]).T.astype(index_dtype)
579 index_dtype = np.dtype(index_type)
584 ]).T.astype(index_dtype)
627 index_dtype = np.dtype(index_type)
655 index_dtype = np.dtype(index_type)
685 index_dtype = np.dtype(index_type)
713 index_dtype = np.dtype(index_type)
743 index_dtype = np.dtype(index_type)
756 indptr = [x.astype(index_dtype) for x in indptr]
[all …]
/dports/databases/arrow/apache-arrow-6.0.1/python/pyarrow/tests/
H A Dtest_serialization.py550 index_dtype = np.dtype(index_type)
555 ]).T.astype(index_dtype)
579 index_dtype = np.dtype(index_type)
584 ]).T.astype(index_dtype)
627 index_dtype = np.dtype(index_type)
655 index_dtype = np.dtype(index_type)
685 index_dtype = np.dtype(index_type)
713 index_dtype = np.dtype(index_type)
743 index_dtype = np.dtype(index_type)
756 indptr = [x.astype(index_dtype) for x in indptr]
[all …]
/dports/security/vault/vault-1.8.2/vendor/github.com/apache/arrow/python/pyarrow/tests/
H A Dtest_serialization.py543 index_dtype = np.dtype(index_type)
548 ]).T.astype(index_dtype)
572 index_dtype = np.dtype(index_type)
577 ]).T.astype(index_dtype)
620 index_dtype = np.dtype(index_type)
648 index_dtype = np.dtype(index_type)
678 index_dtype = np.dtype(index_type)
706 index_dtype = np.dtype(index_type)
736 index_dtype = np.dtype(index_type)
749 indptr = [x.astype(index_dtype) for x in indptr]
[all …]
/dports/misc/py-gluonnlp/gluon-nlp-0.10.0/src/gluonnlp/data/batchify/
H A Dembedding.py68 weight_dtype='float32', index_dtype='int64'): argument
75 self._index_dtype = index_dtype
104 weight_dtype=self._weight_dtype, index_dtype=self._index_dtype)
110 index_dtype): argument
118 self._index_dtype = index_dtype
/dports/devel/py-frictionless/frictionless-4.18.2/frictionless/plugins/
H A Dpandas.py214 index_dtype = self.__write_convert_type(index_field.type)
217 index = index_class(index_rows, name=index_field.name, dtype=index_dtype)
/dports/science/py-scikit-learn/scikit-learn-1.0.2/sklearn/utils/tests/
H A Dtest_sparsefuncs.py891 for index_dtype in [np.int32, np.int64]:
894 if index_dtype is np.int64:
895 X_csr.indptr = X_csr.indptr.astype(index_dtype)
896 X_csr.indices = X_csr.indices.astype(index_dtype)
897 assert X_csr.indices.dtype == index_dtype
898 assert X_csr.indptr.dtype == index_dtype
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/array/
H A Das_strided.py10 index_dtype = {t().itemsize: t for t in np.sctypes['int']} variable
304 dtype=index_dtype[itemsize])
/dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/tests/
H A Dprimitive_harness.py793 index_dtype=np.int32, argument
797 index_dtype = dtypes.canonicalize_dtype(index_dtype)
801 lambda arg: prim.bind(arg, axes=axes, index_dtype=index_dtype), [arr],
805 index_dtype=index_dtype,
818 for index_dtype in jtu.dtypes.all_integer + jtu.dtypes.all_unsigned:
819 _make_argminmax_harness(prim, "index_dtype", index_dtype=index_dtype)
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dlax.py1111 index_dtype: DType) -> Tuple[Array, Array]:
1114 index_dtype=dtypes.canonicalize_dtype(index_dtype))
1117 index_dtype: DType) -> Tuple[Array, Array]:
1120 index_dtype=dtypes.canonicalize_dtype(index_dtype))
5102 def _argminmax_shape_rule(operand, *, axes, index_dtype): argument
5106 def _argminmax_dtype_rule(operand, *, axes, index_dtype): argument
5107 if not dtypes.issubdtype(index_dtype, np.integer):
5109 .format(np.dtype(index_dtype).name))
5110 return index_dtype
5120 index_shape = xc.Shape.array_shape(index_dtype, ())
[all …]
H A Dcontrol_flow.py574 index_dtype = dtypes.result_type(index)
579 if index_dtype.kind not in 'iu':
/dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/
H A Djax2tf.py1448 def _argminmax(fn, operand, axes, index_dtype): argument
1451 if dtypes.iinfo(index_dtype).bits > 32:
1455 return tf.cast(result, to_tf_dtype(index_dtype))
/dports/math/py-jax/jax-0.2.9/jax/_src/numpy/
H A Dlax_numpy.py4110 index_dtype = int64 if use_64bit_index else int32
4111 indices = lax.convert_element_type(indices, index_dtype)
4377 index_dtype = int64 if use_64bit_index else int32
4378 gather_indices = np.zeros((0,), dtype=index_dtype) # use np to save a compilation
4405 lax.convert_element_type(lax.reshape(a, shape + (1,)), index_dtype)
4439 i = lax.convert_element_type(i, index_dtype)
4474 i = lax.convert_element_type(start, index_dtype)
4482 i = arange(start, limit, stride, dtype=index_dtype)