Searched refs:out_types (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | random_choice_op.cc | 67 std::vector<DataType> out_types; in OutputType() local
68 RETURN_IF_NOT_OK(op->OutputType(inputs, out_types)); in OutputType()
69 if (outputs != out_types) { in OutputType()
|
| /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
39 self.func = P.PyFunc(fn, in_types, in_shapes, out_types, out_shapes)
94 def __init__(self, fn, in_types, in_shapes, out_types, out_shapes): argument
96 self.func = P.PyFunc(fn, in_types, in_shapes, out_types, out_shapes)
|
| /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
943 self.add_prim_attr('out_types', out_types)
948 validator.check_value_type("out_types", out_types, [list, tuple], self.name)
950 …validator.check("out_types length", len(out_types), "out_shapes length", len(out_shapes), Rel.EQ, …
964 return tuple(self.out_types)
|
| /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
245 output_infos_.dtypes = out_types; in BuildFuncInfo()
|