| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/src/gluonnlp/utils/ |
| H A D | parameter.py | 174 ignore_extra=False, cast_dtype=None): argument
195 if cast_dtype is not None:
198 cast_dtype=cast_dtype)
|
| /dports/science/py-h5py/h5py-3.6.0/h5py/tests/ |
| H A D | test_dtype.py | 404 cast_dtype=np.uint8
416 cast_dtype=np.dtype([('x', 'u1')])
435 cast_dtype=np.dtype([('x', 'u1'), ('y', [('a', 'u1'), ('b', 'u1')])]),
449 cast_dtype=np.dtype([('x', 'u1'), ('y', '<f8')])
464 cast_dtype=np.dtype([('x', ('?', (3,)))]),
467 def _test_b8(self, arr1, expected_default_cast_dtype, cast_dtype=None): argument
490 if cast_dtype is None:
491 cast_dtype = arr1.dtype
492 arr3 = dset.astype(cast_dtype)[:]
493 self.assertArrayEqual(arr3, arr1.astype(cast_dtype, copy=False))
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/ |
| H A D | parameter.py | 257 def _load_init(self, data, ctx, cast_dtype=False, dtype_source='current'): argument
273 if cast_dtype:
285 if cast_dtype and np.dtype(self.dtype).type != data.dtype:
990 ignore_extra=False, restore_prefix='', cast_dtype=False, argument
1023 ignore_extra, restore_prefix, filename, cast_dtype, dtype_source)
1026 ignore_extra=False, restore_prefix='', filename=None, cast_dtype=False, argument
1071 self[name]._load_init(arg_dict[name], ctx, cast_dtype=cast_dtype,
|
| H A D | block.py | 475 ignore_extra=False, cast_dtype=False, dtype_source='current'): argument
532 cast_dtype=cast_dtype, dtype_source=dtype_source)
557 … params[name]._load_init(loaded[name], ctx, cast_dtype=cast_dtype, dtype_source=dtype_source)
1619 ret.collect_params().load(param_file, ctx, allow_missing, ignore_extra, cast_dtype=True,
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/gluon/ |
| H A D | parameter.py | 257 def _load_init(self, data, ctx, cast_dtype=False, dtype_source='current'): argument
273 if cast_dtype:
285 if cast_dtype and np.dtype(self.dtype).type != data.dtype:
990 ignore_extra=False, restore_prefix='', cast_dtype=False, argument
1023 ignore_extra, restore_prefix, filename, cast_dtype, dtype_source)
1026 ignore_extra=False, restore_prefix='', filename=None, cast_dtype=False, argument
1071 self[name]._load_init(arg_dict[name], ctx, cast_dtype=cast_dtype,
|
| H A D | block.py | 475 ignore_extra=False, cast_dtype=False, dtype_source='current'): argument
532 cast_dtype=cast_dtype, dtype_source=dtype_source)
557 … params[name]._load_init(loaded[name], ctx, cast_dtype=cast_dtype, dtype_source=dtype_source)
1619 ret.collect_params().load(param_file, ctx, allow_missing, ignore_extra, cast_dtype=True,
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/frontend/tflite/ |
| H A D | test_forward.py | 254 def _test_cast(data, cast_dtype): argument
258 out = math_ops.cast(in_data, cast_dtype)
264 _test_cast(np.arange(6.0, dtype=np.float32).reshape((1, 6)), cast_dtype=tf.int32)
265 _test_cast(np.arange(6.0, dtype=np.float32).reshape((1, 6)), cast_dtype=tf.uint8)
266 _test_cast(np.arange(6.0, dtype=np.int32).reshape((1, 6)), cast_dtype=tf.int64)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/tflite/ |
| H A D | test_forward.py | 254 def _test_cast(data, cast_dtype): argument
258 out = math_ops.cast(in_data, cast_dtype)
264 _test_cast(np.arange(6.0, dtype=np.float32).reshape((1, 6)), cast_dtype=tf.int32)
265 _test_cast(np.arange(6.0, dtype=np.float32).reshape((1, 6)), cast_dtype=tf.uint8)
266 _test_cast(np.arange(6.0, dtype=np.int32).reshape((1, 6)), cast_dtype=tf.int64)
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/scripts/instance/mask_rcnn/ |
| H A D | eval_mask_rcnn.py | 143 net.load_parameters(args.pretrained.strip(), cast_dtype=True)
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/scripts/detection/faster_rcnn/ |
| H A D | eval_faster_rcnn.py | 141 net.load_parameters(args.pretrained.strip(), cast_dtype=True)
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/scripts/bert/ |
| H A D | embedding.py | 103 self.bert.load_parameters(params_path, ctx=ctx, ignore_extra=True, cast_dtype=True)
|
| H A D | finetune_classifier.py | 288 cast_dtype=True)
291 nlp.utils.load_parameters(model, model_parameters, ctx=ctx, cast_dtype=True)
|
| H A D | finetune_squad.py | 360 nlp.utils.load_parameters(net, model_parameters, ctx=ctx, cast_dtype=True)
364 ignore_extra=True, cast_dtype=True)
|
| H A D | pretraining_utils.py | 80 nlp.utils.load_parameters(model, param_path, ctx=ctx, cast_dtype=True)
|
| /dports/math/py-pandas/pandas-1.2.5/pandas/tests/ |
| H A D | test_nanops.py | 128 cast_dtype = "c16" if has_c16 else "f8"
129 res = res.astype(cast_dtype)
130 targ = targ.astype(cast_dtype)
|
| /dports/misc/py-gluonnlp/gluon-nlp-0.10.0/scripts/language_model/ |
| H A D | run_squad.py | 212 cast_dtype=True)
226 nlp.utils.load_parameters(net, args.model_parameters, ctx=ctx, cast_dtype=True)
|
| H A D | run_glue.py | 386 cast_dtype=True)
|
| /dports/misc/py-gluoncv/gluon-cv-0.9.0/gluoncv/model_zoo/ |
| H A D | googlenet.py | 342 tag=pretrained, root=root), ctx=ctx, cast_dtype=True)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/tflite/ |
| H A D | test_forward.py | 626 def _test_cast(data, cast_dtype): argument
630 out = math_ops.cast(in_data, cast_dtype)
636 _test_cast(np.arange(6.0, dtype=np.float32).reshape((1, 6)), cast_dtype=tf.int32)
637 _test_cast(np.arange(6.0, dtype=np.float32).reshape((1, 6)), cast_dtype=tf.uint8)
638 _test_cast(np.arange(6.0, dtype=np.int32).reshape((1, 6)), cast_dtype=tf.int64)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/frontend/ |
| H A D | tflite.py | 1080 cast_dtype = cast_options.OutDataType()
1082 out = _op.cast(in_expr, self.get_tensor_type_str(cast_dtype))
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/frontend/ |
| H A D | tflite.py | 1080 cast_dtype = cast_options.OutDataType()
1082 out = _op.cast(in_expr, self.get_tensor_type_str(cast_dtype))
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/ |
| H A D | quantization.py | 1002 net.collect_params().load(param_name, cast_dtype=True, dtype_source='saved')
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/ |
| H A D | quantization.py | 1002 net.collect_params().load(param_name, cast_dtype=True, dtype_source='saved')
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_gluon.py | 170 params1.load('test_parameter_dict.params', cast_dtype=True, dtype_source='current')
173 params1.load('test_parameter_dict.params', cast_dtype=True, dtype_source='saved')
3179 net.load_parameters('test_gluon_param_load.params', cast_dtype=True)
3189 …net.load_parameters('test_gluon_param_load_dtype_source.params', cast_dtype=True, dtype_source="sa…
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
| H A D | test_gluon.py | 170 params1.load('test_parameter_dict.params', cast_dtype=True, dtype_source='current')
173 params1.load('test_parameter_dict.params', cast_dtype=True, dtype_source='saved')
3179 net.load_parameters('test_gluon_param_load.params', cast_dtype=True)
3189 …net.load_parameters('test_gluon_param_load_dtype_source.params', cast_dtype=True, dtype_source="sa…
|