Home
last modified time | relevance | path

Searched defs:x_tensor (Results 1 – 25 of 123) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/customize/
Didentity.cc25 …CustomizeCallWithoutContigous(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in IdentityCustomizeCallWithoutContigous()
61 void IdentityCustomizeCall(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in IdentityCustomizeCall()
87 …seTensorPtr IdentityCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in IdentityCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dneg.cc23 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
28 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Delu_ext.cc33 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
39 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
Dfloor.cc33 const auto &x_tensor = input_values[0]->cast<tensor::BaseTensorPtr>(); in InferType() local
39 const auto &x_tensor = input_values[0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Drelu.cc39 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
46 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dleaky_relu_grad_ext.cc36 const auto &x_tensor = input_values[kIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
41 const auto &x_tensor = input_values[kIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Duniform_ext.cc47 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
53 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dsoftplus_ext.cc46 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
53 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dclamp_scalar.cc46 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
68 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dleaky_relu_ext.cc39 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
48 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dgelu.cc48 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
56 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Disfinite.cc47 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
56 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dcast.cc50 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
61 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Deltwise_op.cc49 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
56 const auto &x_tensor = input_values[kInputIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
Dceil.cc49 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
58 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dsqrt.cc57 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
73 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Drsqrt.cc48 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
64 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Derfinv.cc44 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
59 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dnon_zero.cc58 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
70 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
Derf.cc43 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
58 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
Dlog.cc84 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
90 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
104 auto x_tensor = x_value->cast<tensor::TensorPtr>(); in InferValue() local
Dtanh.cc47 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferShape() local
53 const auto &x_tensor = input_values[kIndex0]->cast<tensor::BaseTensorPtr>(); in InferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/
Dsilu.cc28 OpPtr SiLUCPUCall(const device::DeviceContext *device_context, const BaseTensorPtr &x_tensor) { in SiLUCPUCall()
39 void SiLUCPUCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in SiLUCPUCustomize()
Dsilu_grad.cc31 const BaseTensorPtr &x_tensor) { in SiLUGradCPUCall()
52 const BaseTensorPtr &x_tensor) { in SiLUGradCPUCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Didentity.cc38 …CustomizeCallWithoutContigous(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in IdentityCustomizeCallWithoutContigous()
97 void IdentityCustomizeCall(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in IdentityCustomizeCall()
142 …orPtr IdentityAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor) { in IdentityAscendCustomize()

12345