Searched refs:out_dtype (Results 1 – 8 of 8) sorted by relevance
| /third_party/boost/boost/python/numpy/ |
| D | ufunc.hpp | 107 dtype out_dtype = dtype::get_builtin<TResult>(); in call() local
110 from_object(output, out_dtype, ndarray::ALIGNED | ndarray::WRITEABLE) in call()
111 : zeros(in_array.get_nd(), in_array.get_shape(), out_dtype); in call()
170 dtype out_dtype = dtype::get_builtin<TResult>(); in call() local
175 ? from_object(output, out_dtype, ndarray::ALIGNED | ndarray::WRITEABLE) in call()
176 : zeros(iter.get_nd(), iter.get_shape(), out_dtype); in call()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | fft_real.cc | 35 auto out_dtype = kFloat32; in FftRealInfer() local
38 …return std::make_shared<abstract::AbstractTensor>(out_dtype, std::make_shared<abstract::Shape>(out… in FftRealInfer()
|
| /third_party/mindspore/mindspore/lite/test/st/scripts/ |
| D | base_functions.sh | 7 quant_type bit_num config_file train_model in_dtype out_dtype converter_result cfg_file
51 out_dtype="DEFAULT"
66 out_dtype="FLOAT"
71 ' --inputDataType='${in_dtype}' --outputDataType='${out_dtype}' \
74 --inputDataType=${in_dtype} --outputDataType=${out_dtype} \
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 216 out_dtype = (hashmap_dtype, hashmap_dtype,
220 'dtype': out_dtype,
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | model_builder.py | 105 out_shape, out_dtype, out_format = op_infer.infer(prim, tensor_inputs, attrs)
106 output = self.tensor(out_shape, out_dtype, out_format, name)
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 551 auto out_dtype = output[ATTR_DTYPE]; in PyInferRes2Abstract() local
554 return MakePyInferRes2Abstract(out_shape, out_dtype, output); in PyInferRes2Abstract()
558 TypePtr dtype = py::isinstance<Type>(out_dtype) ? out_dtype.cast<TypePtr>() : nullptr; in PyInferRes2Abstract()
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_math_ops.py | 156 def infer_dtype(self, x_dtype, out_dtype, dout_dtype): argument
|
| D | test_nn_ops.py | 128 def infer_dtype(self, x_dtype, out_dtype, dout_dtype): argument
|