Home
last modified time | relevance | path

Searched defs:input1_shape (Results 1 – 25 of 48) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Darithmetic_tests.cc45 const std::vector<int> &input1_shape, in CreateParameter()
60 std::vector<int> input1_shape = {1, 2, 2, 3}; in TEST_F() local
75 std::vector<int> input1_shape = {1}; in TEST_F() local
89 std::vector<int> input1_shape = {3}; in TEST_F() local
104 std::vector<int> input1_shape = {1, 2, 2, 3}; in TEST_F() local
118 std::vector<int> input1_shape = {2, 2, 2, 3}; in TEST_F() local
132 std::vector<int> input1_shape = {1, 1, 1, 4}; in TEST_F() local
147 std::vector<int> input1_shape = {1, 1, 3, 4}; in TEST_F() local
162 std::vector<int> input1_shape = {1, 2, 2, 3}; in TEST_F() local
Dconcat_tests.cc34 std::vector<int> input1_shape = {1, 1, 1, 8}; in TEST_F() local
49 std::vector<int> input1_shape = {1}; in TEST_F() local
64 std::vector<int> input1_shape = {2, 2, 2, 8}; in TEST_F() local
92 std::vector<int> input1_shape = {2, 3, 2, 8}; in TEST_F() local
187 std::vector<int> input1_shape = {1, 1, 8}; in TEST_F() local
218 std::vector<int> input1_shape = {2, 2, 2, 9}; in TEST_F() local
249 std::vector<int> input1_shape = {1, 7}; in TEST_F() local
267 std::vector<int> input1_shape = {1, 1, 1, 7}; in TEST_F() local
290 std::vector<int> input1_shape = {1, 1, 1, 7}; in TEST_F() local
317 std::vector<int> input1_shape = {1, 1, 1, 7}; in TEST_F() local
Dpower_tests.cc36 std::vector<int> input1_shape = {1, 2, 8}; in TEST_F() local
53 std::vector<int> input1_shape = {2, 8}; in TEST_F() local
72 std::vector<int> input1_shape = {2, 7}; in TEST_F() local
104 std::vector<int> input1_shape = {2, 8}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcombined_non_max_suppression.cc57 …edNonMaxSuppressionCheckShapeSize(const ShapeVector &input0_shape, const ShapeVector &input1_shape, in CombinedNonMaxSuppressionCheckShapeSize()
75 …dNonMaxSuppressionCheckShapeValue(const ShapeVector &input0_shape, const ShapeVector &input1_shape, in CombinedNonMaxSuppressionCheckShapeValue()
103 …auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetS… in CombinedNonMaxSuppressionGetOutputShape() local
145 const ShapeVector &input1_shape, bool is_dynamic) { in CheckCombinedNonMaxSuppressionThreshold()
168 …auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->GetS… in CombinedNonMaxSuppressionInferShape() local
Dsequence_add.cc70 auto input1_shape = input1->GetShape()->cast<abstract::SequenceShapePtr>(); in SequenceAddInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dtridiagonalsolve_cpu_kernel.h57 ShapeVector input1_shape; variable
Dheaviside_cpu_kernel.h59 ShapeVector input1_shape; variable
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dfloor_div.cc27 auto input1_shape = input_args[kIndex1]->GetShape(); in InferShape() local
Dfloor_mod.cc27 auto input1_shape = input_args[kIndex1]->GetShape(); in InferShape() local
Dclamp_tensor.cc66 auto input1_shape = input_args[kInputIndex1]->GetShape(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dlin_space_gpu_kernel.cc45 auto input1_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
Dcombined_non_max_suppression_gpu_kernel.cc69 std::vector<size_t> input1_shape = in Resize() local
Dmatmul_gpu_kernel.cc116 auto input1_shape = Convert2SizeTClipNeg(input1_shape_signed); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dbounding_box_encode_info.cc72 Shape input1_shape = inputs_shape_[1]; in GenerateOpStrategies() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dmatmul.cc101 auto input1_shape = input1_tensor_shape->GetDimSizes(); in MatMulCompute() local
Dmaximum_grad_grad.cc95 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in MaximumGradGradCompute() local
Dminimum_grad_grad.cc97 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in MinimumGradGradCompute() local
Dmatrix_solve.cc93 auto input1_shape = input1_tensor_shape->GetDimSizes(); in MatrixSolveCompute() local
Ddiv.cc338 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in DivComputeInt() local
370 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in DivCompute() local
Dlogical_xor.cc183 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in LogicalXorCompute() local
Dbatchmatmul.cc90 auto input1_shape = input1_tensor_shape->GetDimSizes(); in DoCompute() local
Dxdivy.cc176 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in XdivyCompute() local
Dmul_no_nan.cc235 auto input1_shape = input1_tensor->GetTensorShape()->GetDimSizes(); in MulNoNanCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dscale_int8.cc103 auto input1_shape = in_tensors_[SECOND_INPUT]->shape(); in InitParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dtridiagonal_matmul_cpu_kernel.cc61 auto input1_shape = inputs[kIndex1]->GetShapeVector(); in Resize() local

12