Home
last modified time | relevance | path

Searched refs:infer_value (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ops/operations/
Dmath_ops.py193 def infer_value(self, x, y): member in Add
215 def infer_value(self, x, y): member in TensorAdd
1058 def infer_value(self, x, axis): member in CumProd
1520 def infer_value(self, input_x): member in Neg
1707 def infer_value(self, x, y): member in Sub
1758 def infer_value(self, x, y): member in Mul
1890 def infer_value(self, x): member in Rsqrt
1937 def infer_value(self, x): member in Sqrt
1992 def infer_value(self, x): member in Reciprocal
2049 def infer_value(self, x, power): member in Pow
[all …]
Darray_ops.py1450 def infer_value(self, x): member in TupleToArray
1505 def infer_value(self, x): member in ScalarToArray
1544 def infer_value(self, x, dtype=mstype.float32): member in ScalarToTensor
3405 def infer_value(self, x): member in Diag
3469 def infer_value(self, x): member in DiagPart
3528 def infer_value(self, n, m, t): member in Eye
6035 def infer_value(self, start_value, limit_value, delat_value): member in Range
D_inner_ops.py174 def infer_value(self, x_value): member in Range
697 def infer_value(self, input_tensor): member in ErrorOnDynamicShapeInput
Dcomm_ops.py927 def infer_value(self, x, dev_mat, tensor_map): member in _GetTensorSlice
Dnn_ops.py3147 def infer_value(self, var, mean_square, moment, learning_rate, grad, decay, momentum, epsilon): member in ApplyRMSProp
/third_party/mindspore/mindspore/core/ops/
Dskip_gram.cc29 auto infer_value = input_args[0]->BuildValue(); in InferShape() local
30 if (infer_value == nullptr) { in InferShape()
/third_party/mindspore/mindspore/ops/
Dprimitive.py496 def infer_value(self, *args): member in PrimitiveWithInfer
664 def infer_value(self, *args): member in constexpr.deco.CompileOp
/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.cc388 auto infer_value = python_obj_.attr(PY_PRIM_METHOD_INFER_VALUE); in RunInferValue() local
389 return infer_value(*args); in RunInferValue()
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc797 int64_t infer_value = x_num; in InferImplReshape() local
803 infer_value = infer_value / value; in InferImplReshape()
808 shape[index] = infer_value; in InferImplReshape()