| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sequence_less.cc | 82 auto x_input = input_args[x_index]; in SequenceLessInferInner() local
106 auto x_input = input_args[kIndex0]; in InferShape() local
118 auto x_input = input_args[kIndex0]; in InferType() local
139 auto x_input = input_args[kIndex0]; in InferShape() local
151 auto x_input = input_args[kIndex0]; in InferType() local
|
| D | sequence_greater.cc | 97 auto x_input = input_args[kIndex0]; in InferShape() local
109 auto x_input = input_args[kIndex0]; in InferType() local
130 auto x_input = input_args[kIndex0]; in InferShape() local
142 auto x_input = input_args[kIndex0]; in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | elementwise_tensorrt.cc | 99 void ElementWiseTensorRT::LogicalOpChangeInputType(TensorRTContext *ctx, ITensorHelper *x_input, in LogicalOpChangeInputType()
114 ITensorHelper x_input; in AddInnerOp() local
182 int ElementWiseTensorRT::PreprocessInputTensors(TensorRTContext *ctx, ITensorHelper *x_input, ITens… in PreprocessInputTensors()
226 int ElementWiseTensorRT::BroadcastInputTensors(TensorRTContext *ctx, ITensorHelper *x_input, ITenso… in BroadcastInputTensors()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sync_batch_norm_grad_impl.cu | 30 __global__ void SyncBatchNormGradPre(size_t N, size_t C, size_t H, size_t W, const T *x_input, cons… in SyncBatchNormGradPre()
95 __global__ void SyncBatchNormGradPost(size_t N, size_t C, size_t H, size_t W, const T *x_input, con… in SyncBatchNormGradPost()
131 cudaError_t CalSyncBatchNormGradPre(size_t N, size_t C, size_t H, size_t W, const T *x_input, const… in CalSyncBatchNormGradPre()
139 cudaError_t CalSyncBatchNormGradPost(size_t N, size_t C, size_t H, size_t W, const T *x_input, cons… in CalSyncBatchNormGradPost()
|
| D | batchnorm_grad_impl.cu | 31 __global__ void BatchNormGradKernel(T *x_input, T *dy, float *scale, float *save_mean, float *save_… in BatchNormGradKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | batchnorm_fold.py | 61 def _batchnorm_fold_compute(x_input, x_sum, x_square_sum, mean, variance, momentum, epsilon): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | matmul_biasadd_fusion.cc | 72 auto x_input = utils::cast<AnfNodePtr>((*equiv)[x_]); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | mc2_fusion.cc | 157 auto x_input = std::make_shared<Var>(); // input x in DefineFusionPattern() local
233 auto x_input = std::make_shared<Var>(); // input x in DefineFusionPattern() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | unsorted_segment_sum_fission.cc | 37 auto x_input = sum->input(kIndex1); in CreateConcatD() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_sequence/ |
| D | test_dynamic_sequence_unstack.py | 86 def input_grad_comp(x_input, net, out_grad_np): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | matmul_allreduce_fusion.cc | 109 auto x_input = std::make_shared<Var>(); // input x in DefineQuantBatchMatmulAllReducePattern() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | selection_ops_proto.cc | 31 TensorDesc x_input = op.GetInputDesc(0); in InferShapeAndTypeMaskedSelect() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 3593 auto x_input = node->input(kOneNum); in ExportPrimTensorCopySlices() local
3785 auto x_input = node->input(kOneNum); in ExportPrimSort() local
|