Searched refs:input_signature (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/common/ |
| D | api.py | 125 def __init__(self, fn, input_signature=None, obj=None): argument
127 self.input_signature = input_signature
149 if self.input_signature is not None:
151 for sig_spec in self.input_signature:
225 def ms_function(fn=None, obj=None, input_signature=None): argument
290 out = _MindsporeFunctionExecutor(func, input_signature, process_obj)(*args)
|
| /third_party/mesa3d/src/microsoft/compiler/ |
| D | dxil_signature.c | 617 const struct dxil_mdnode *input_signature = get_input_signature(mod, s, vulkan); in get_signatures() local
621 input_signature, in get_signatures()
625 if (output_signature || input_signature) in get_signatures()
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_staging.py | 73 @ms_function(input_signature=(MetaTensor(dtype.float32, (1, 1, 3, 3)),
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.h | 148 py::bool_ VerifyInputSignature(const py::list &input_signature, const py::tuple &inputs);
|
| D | pipeline.cc | 273 py::bool_ VerifyInputSignature(const py::list &input_signature, const py::tuple &inputs) { in VerifyInputSignature() argument
275 if (inputs.size() != input_signature.size()) { in VerifyInputSignature()
289 auto sig = input_signature[count].cast<std::shared_ptr<MetaTensor>>(); in VerifyInputSignature()
|