Home
last modified time | relevance | path

Searched defs:out_shapes (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/tests/st/ops/gpu/
Dtest_pyfunc_op.py37 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/
Drandom_choice_op.cc52 std::vector<TensorShape> out_shapes; in OutputShape() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dget_next_info.cc206 Shapes out_shapes; in InferReplaceOps() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/
Dpy_func_cpu_kernel.cc232 …const auto &out_shapes = AnfAlgo::GetNodeAttr<std::vector<std::vector<int64_t>>>(kernel_node, "out… in BuildFuncInfo() local
/third_party/mindspore/mindspore/ops/operations/
Dother_ops.py937 def __init__(self, fn, in_types, in_shapes, out_types, out_shapes, stateful=True): argument