Home
last modified time | relevance | path

Searched refs:scalar_to_tensor (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ops/_grad_experimental/
Dgrad_math_ops.py93 root_pi_over_two = cast(sqrt(F.scalar_to_tensor(np.pi)) / 2, dtype(dout))
/third_party/mindspore/mindspore/ops/
Dfunctional.py126 scalar_to_tensor = P.ScalarToTensor() variable
/third_party/mindspore/mindspore/ops/composite/multitype_ops/
D_compile_utils.py422 tensor_index = F.scalar_to_tensor(int_index, mstype.int64)
503 tensor_index = F.scalar_to_tensor(int_index, mstype.int64)
841 result = const_utils.scalar_to_tensor(False)
D_constexpr_utils.py644 def scalar_to_tensor(x): function
/third_party/mindspore/tests/ut/python/ops/
Dtest_nn_ops_check.py56 'desc_inputs': [F.scalar_to_tensor(5.0)],
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_math_ops.py523 half_root_pi = cast(2 / sqrt(F.scalar_to_tensor(np.pi)), dtype(x))
541 half_root_pi = cast(2 / sqrt(F.scalar_to_tensor(np.pi)), dtype(x))
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dparse.cc1423 auto scalar_to_tensor = prim::GetPythonOps("ScalarToTensor", "mindspore.ops.operations"); in ParseForLoop() local
1424 …auto scalar_to_tensor_node = block->func_graph()->NewCNodeInOrder({NewValueNode(scalar_to_tensor)}… in ParseForLoop()
1448 …auto body_scalar_to_tensor_node = body_func_graph->NewCNodeInOrder({NewValueNode(scalar_to_tensor)… in ParseForLoop()