Searched defs:in_shapes (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 | compose_op.cc | 26 std::vector<TensorShape> in_shapes = inputs; in OutputShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/ |
| D | py_func_cpu_kernel.cc | 230 …const auto &in_shapes = AnfAlgo::GetNodeAttr<std::vector<std::vector<int64_t>>>(kernel_node, "in_s… 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
|