| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | sigmoid_grad.cc | 28 const BaseTensorPtr &dy_tensor, const BaseTensorPtr &y_tensor, in SigmoidGradAscendCall()
34 …rPtr SigmoidGradAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &y_tensor, in SigmoidGradAscendCustomize()
|
| D | tanh_grad.cc | 28 const BaseTensorPtr &dy_tensor, const BaseTensorPtr &y_tensor, in TanhGradAscendCall()
34 …nsorPtr TanhGradAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &y_tensor, in TanhGradAscendCustomize()
|
| D | relu_grad.cc | 28 … const BaseTensorPtr &dy_tensor, const BaseTensorPtr &y_tensor, const ScalarPtr &threshold, in ReluGradAscendCall()
35 const BaseTensorPtr &y_tensor) { in ReluGradAscendCustomize()
|
| D | sub.cc | 28 const BaseTensorPtr &y_tensor) { in SubAscendCustomize()
|
| D | add.cc | 28 const BaseTensorPtr &y_tensor) { in AddAscendCustomize()
|
| D | gelu_grad.cc | 27 … const BaseTensorPtr &x_tensor, const BaseTensorPtr &y_tensor) { in GeLUGradAscendCustomize()
|
| D | divmod.cc | 28 const BaseTensorPtr &y_tensor, in DivModAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/customize/ |
| D | divmod.cc | 27 …onst std::shared_ptr<OpRunner> &op, const BaseTensorPtr &x_tensor, const BaseTensorPtr &y_tensor) { in FloorDivCall()
78 … const BaseTensorPtr &y_tensor, const std::optional<Int64ImmPtr> &rounding_mode) { in DivModCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | bmm_ext.cc | 105 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
122 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| D | dense.cc | 117 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
144 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| D | matmul.cc | 140 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
171 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| D | matmul_ext.cc | 180 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
197 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| D | batch_mat_mul.cc | 185 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferType() local
216 const auto &y_tensor = input_values[kInputIndex1]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| D | addmm.cc | 90 const auto &y_tensor = input_values[kIndex2]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| D | select.cc | 73 const auto &y_tensor = input_values[kSelectYIndex]->cast<tensor::BaseTensorPtr>(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | divmod.cc | 25 const BaseTensorPtr &y_tensor, in DivModCPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/ |
| D | select.cc | 70 const tensor::TensorPtr &x_tensor, const tensor::TensorPtr &y_tensor, in SelectInnerInferValue()
104 auto y_tensor = y->cast<tensor::TensorPtr>(); in InferValue() local
|
| D | real_div.cc | 93 auto y_tensor = y->cast<tensor::TensorPtr>(); in InferValue() local
|
| D | div.cc | 93 auto y_tensor = y->cast<tensor::TensorPtr>(); in InferValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/ |
| D | divmod.cc | 26 const BaseTensorPtr &y_tensor, in DivModGPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | cauchy.cc | 71 Tensor *y_tensor = ctx.Output(0); in CauchyCompute() local
|
| D | tridiagonal_matmul.cc | 117 auto y_tensor = ctx.Output(0); in TridiagonalMatMulCompute() local
|
| D | hamming_window.cc | 105 Tensor *y_tensor = ctx.Output(0); in HammingWindowCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | softmax_crossentropy_fp32_tests.cc | 41 lite::Tensor y_tensor(TypeId::kNumberTypeFloat32, dim_y); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_notequal.cc | 83 …auto y_tensor = std::make_shared<tensor::Tensor>(param.y_type, param.y_shape, (void *)¶m.y_dat… in TEST_P() local
|