Searched defs:input_signature (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/train/ |
| D | data_sink.py | 29 def _init_sink_dataset(dataset, sink_size, input_signature, create_info): argument
128 def data_sink(fn, dataset, sink_size=1, jit_config=None, input_signature=None): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/train/ |
| D | test_data_sink.py | 76 def _train_func_sink(model, dataset, loss_fn, opt, input_signature=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | auto_dynamic_shape.py | 496 def get_auto_dynamic_shape_args_with_check_input_signature(compile_args, key_id, input_signature): argument
503 …_auto_dynamic_shape_phase_with_check_input_signature(compile_args, key_id, phase, input_signature): argument
|
| D | api.py | 534 def __init__(self, fn, ms_create_time, input_signature=None, obj=None, jit_config=None): argument
798 def jit(fn=None, mode="PSJit", input_signature=None, hash_args=None, jit_config=None, compile_once=… argument
986 def ms_function(fn=None, input_signature=None, hash_args=None, jit_config=None): argument
|
| /third_party/mesa3d/src/microsoft/compiler/ |
| D | dxil_signature.c | 883 …const struct dxil_mdnode *input_signature = get_signature_metadata(mod, mod->inputs, mod->psv_inpu… in get_signatures() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/ |
| D | pipeline.cc | 713 py::bool_ VerifyInputSignature(const py::list &input_signature, const py::tuple &inputs) { in VerifyInputSignature()
|