Searched defs:out_shapes (Results 1 – 5 of 5) 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 | 52 std::vector<TensorShape> out_shapes; in OutputShape() local
|
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
D | get_next_info.cc | 206 Shapes out_shapes; in InferReplaceOps() local
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/ |
D | py_func_cpu_kernel.cc | 232 …const auto &out_shapes = AnfAlgo::GetNodeAttr<std::vector<std::vector<int64_t>>>(kernel_node, "out… 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
|