Searched defs:out_types (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_pyfunc_op.py | 37 def __init__(self, fn, in_types, in_shapes, out_types, out_shapes): argument
94 def __init__(self, fn, in_types, in_shapes, out_types, out_shapes): argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | random_choice_op.cc | 67 std::vector<DataType> out_types; in OutputType() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/ |
| D | py_func_cpu_kernel.cc | 233 const auto &out_types = AnfAlgo::GetNodeAttr<std::vector<TypePtr>>(kernel_node, "out_types"); in BuildFuncInfo() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | other_ops.py | 937 def __init__(self, fn, in_types, in_shapes, out_types, out_shapes, stateful=True): argument
|