Home
last modified time | relevance | path

Searched defs:SetTensorMixPrecisionCast (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/
Ddo_pyboost_cast.cc68 ValuePtr PyBoostCastOperation::SetTensorMixPrecisionCast(const FrontendOpRunInfoPtr &op_run_info, c… in SetTensorMixPrecisionCast() function in mindspore::pynative::PyBoostCastOperation
77 tensor::BaseTensorPtr PyBoostCastOperation::SetTensorMixPrecisionCast(const FrontendOpRunInfoPtr &o… in SetTensorMixPrecisionCast() function in mindspore::pynative::PyBoostCastOperation
102 std::optional<tensor::BaseTensorPtr> PyBoostCastOperation::SetTensorMixPrecisionCast( in SetTensorMixPrecisionCast() function in mindspore::pynative::PyBoostCastOperation
111 ValueTuplePtr PyBoostCastOperation::SetTensorMixPrecisionCast(const FrontendOpRunInfoPtr &op_run_in… in SetTensorMixPrecisionCast() function in mindspore::pynative::PyBoostCastOperation
116 ValueListPtr PyBoostCastOperation::SetTensorMixPrecisionCast(const FrontendOpRunInfoPtr &op_run_inf… in SetTensorMixPrecisionCast() function in mindspore::pynative::PyBoostCastOperation
Ddo_pyboost_cast.h57 …T SetTensorMixPrecisionCast(const FrontendOpRunInfoPtr &op_run_info, const T &t, size_t index) con… in SetTensorMixPrecisionCast() function
Ddo_cast.cc246 void CastOperation::SetTensorMixPrecisionCast(const FrontendOpRunInfoPtr &op_run_info) const { in SetTensorMixPrecisionCast() function in mindspore::pynative::CastOperation