Searched defs:input_x_tensor (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | tile.cc | 28 void TileAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_x_tensor, in TileAscendCustomize()
|
| D | flatten_ext.cc | 30 …FlattenExtAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_x_tensor, in FlattenExtAscendCustomize()
|
| D | argmax_ext.cc | 28 …Ptr ArgMaxAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_x_tensor, in ArgMaxAscendCustomize()
|
| D | grid_sampler_2d_grad.cc | 31 …d::shared_ptr<OpRunner> &op, const BaseTensorPtr &grad_tensor, const BaseTensorPtr &input_x_tensor, in GridSampler2DGradAscendCustomize()
|
| D | norm.cc | 30 void NormAscendCustomize(const std::shared_ptr<OpRunner> &op, const BaseTensorPtr &input_x_tensor, in NormAscendCustomize()
|
| D | grid_sampler_3d_grad.cc | 31 …d::shared_ptr<OpRunner> &op, const BaseTensorPtr &grad_tensor, const BaseTensorPtr &input_x_tensor, in GridSampler3DGradAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | logit_grad.cc | 68 auto input_x_tensor = ctx.Input(1); in LogitGradCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/dynamic_obfuscation/ |
| D | dynamic_obfuscation.cc | 691 tensor::TensorPtr input_x_tensor = std::make_shared<Tensor>(x_type_id, x_shape); in CloneSubGraph() local
764 tensor::TensorPtr input_x_tensor = std::make_shared<Tensor>(x_type_id, x_shape); in BuildFakeGraph() local
|