Home
last modified time | relevance | path

Searched refs:broadcast_to (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/tests/st/ops/cpu/
Dtest_broadcast_to_op.py33 expect = np.broadcast_to(x_np, shape)
39 expect = np.broadcast_to(x_np, shape)
45 expect = np.broadcast_to(x_np, shape)
51 expect = np.broadcast_to(x_np, shape)
57 expect = np.broadcast_to(x_np, shape)
62 expect = np.broadcast_to(x1_np, shape)
68 expect = np.broadcast_to(x1_np, shape)
74 expect = np.broadcast_to(x1_np, shape)
91 expect = np.broadcast_to(x_np, np_shape)
96 expect = np.broadcast_to(x1_np, np_shape)
[all …]
/third_party/mindspore/tests/st/ops/gpu/
Dtest_broadcast_to_ops.py33 expect = np.broadcast_to(x_np, shape)
38 expect = np.broadcast_to(x1_np, shape)
44 expect = np.broadcast_to(x1_np, shape)
61 expect = np.broadcast_to(x_np, np_shape)
66 expect = np.broadcast_to(x1_np, np_shape)
73 expect = np.broadcast_to(x1_np, np_shape)
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dcdist_fission.cc74 auto broadcast_to = func_graph->NewCNode(broadcast_to_inputs); in AddBroadCastToNode() local
75 AnfAlgo::SetOutputInferTypeAndShape({dtype}, {need_shape}, broadcast_to.get()); in AddBroadCastToNode()
78 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(shape), broadcast_to); in AddBroadCastToNode()
79 AnfAlgo::SetNodeAttr("is_backend_insert", MakeValue(true), broadcast_to); in AddBroadCastToNode()
80 return broadcast_to; in AddBroadCastToNode()
/third_party/mindspore/mindspore/nn/probability/distribution/_utils/
D__init__.py18 from .custom_ops import exp_generic, log_generic, broadcast_to
Dcustom_ops.py71 def broadcast_to(x, target): function
/third_party/mindspore/mindspore/numpy/
Darray_ops.py1246 def broadcast_to(array, shape): function
1315 res.append(broadcast_to(arr, out_shape))
2172 tmp.append(broadcast_to(choice, shape_choice))
2178 choices = broadcast_to(choices, (F.shape(choices)[0],) + shape_choice)
2182 a = broadcast_to(a, shape_choice)
D__init__.py32 tile, broadcast_to, broadcast_arrays, roll, append, split, vsplit,
Darray_creations.py38 from .array_ops import ravel, concatenate, broadcast_arrays, reshape, broadcast_to, flip, \
376 return broadcast_to(fill_value, shape)
/third_party/mindspore/mindspore/nn/probability/distribution/
Dcategorical.py26 from ._utils.custom_ops import exp_generic, log_generic, broadcast_to
134 self.broadcast = broadcast_to
Ddistribution.py23 from ._utils.custom_ops import broadcast_to, exp_generic, log_generic
101 self.broadcast = broadcast_to
/third_party/mindspore/mindspore/ops/composite/multitype_ops/
D_compile_utils.py977 broadcast_to = P.BroadcastTo(shape_out)
979 where = broadcast_to(where)
981 a = broadcast_to(a)
982 initial = broadcast_to(initial)
/third_party/mindspore/mindspore/ops/_op_impl/tbe/
D__init__.py341 from .broadcast_to import _broadcast_to_tbe
/third_party/mindspore/tests/st/numpy_native/
Dtest_array_ops.py882 a = mnp.broadcast_to(x, (2, 3))
883 b = mnp.broadcast_to(x, (8, 1, 3))
888 a = onp.broadcast_to(x, (2, 3))
889 b = onp.broadcast_to(x, (8, 1, 3))
/third_party/mindspore/mindspore/_extends/parse/
Dstandard_method.py1498 broadcast_to = P.BroadcastTo(shape_(y))
1499 return broadcast_to(x)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
DBUILD.gn24 "base/broadcast_to.c",