Home
last modified time | relevance | path

Searched defs:input_1 (Results 1 – 25 of 54) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dinsert_type_transform_op_test.py84 def before(input_1, input_2, x): argument
90 def after(input_1, input_2, x): argument
110 def before(input_1, input_2, input_3): argument
116 def after(input_1, input_2, input_3): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsequence_add.cc55 std::shared_ptr<T> GetOutputType(const AbstractBasePtr &input_1, const AbstractBasePtr &input_2) { in GetOutputType()
84 const auto &input_1 = input_args[kIndex0]; in SequenceAddInferType() local
100 auto input_1 = dyn_cast<abstract::AbstractSequence>(input_args[input_1_index]); in SequenceAddInferInner() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dlogical_xor.cc52 Tensor *input_1 = ctx.Input(1); in LogicalXorCheck() local
103 auto input_1 = reinterpret_cast<T *>(ctx.Input(1)->GetData()); in NoBcastCompute() local
142 auto input_1 = reinterpret_cast<T *>(ctx.Input(1)->GetData()); in BcastCompute() local
Dmultinomial.cc53 uint32_t Generate(Tensor *input_0, Tensor *input_1, Tensor *output, CpuKernelContext &ctx) { in Generate()
170 Tensor *input_1 = ctx.Input(kSecondInputIndex); in Compute() local
Drandom_poisson.cc89 Tensor *input_1 = ctx.Input(kSecondInputIndex); in Generate() local
Dleft_shift.cc72 Tensor *input_1 = ctx.Input(1); in LeftShiftParamCheck() local
165 auto input_1 = static_cast<T *>(ctx.Input(1)->GetData()); in BcastCompute() local
Dlu_solve.cc41 Tensor *input_1 = ctx.Input(1); in Compute() local
136 T *input_1 = new T[input_1_Shape->NumElements()]; in LuSolveCompute() local
Dselect.cc74 Tensor *input_1 = ctx.Input(1); in SelectParamCheck() local
Dbias_add.cc40 Tensor *input_1 = ctx.Input(kSecondInputIndex); in Compute() local
Drsqrt_grad.cc37 Tensor *input_1 = ctx.Input(kSecondInputIndex); in Compute() local
Dsparse_reshape.cc55 Tensor *input_1 = ctx.Input(1); in Compute() local
Dbetainc.cc106 Eigen::TensorMap<Eigen::Tensor<T, 1>> input_1(data_pointers[1] + start, end - start); in RunParallel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dresize_grad_infer.c40 const TensorC *input_1 = inputs[1]; in ResizeGradInferShape() local
/third_party/mindspore/mindspore-src/source/tests/st/graph_syntax/
Dtest_expr_clear.py41 def construct(self, input_1, input_2): argument
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_sequence_add.cc47 auto input_1 = std::make_shared<abstract::AbstractTuple>(list1); in compare_tuple() local
62 auto input_1 = std::make_shared<abstract::AbstractList>(list1); in compare_list() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/infer/
Ddpico_roi_align_infer.cc107 const auto &input_1 = (*inputs)[1]; in Infer() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/fusion_inout_test/
Dtrans_matmul_fusion_inout_test.cc42 …auto input_1 = AddParameter(graph_, 0, {1, kHeight, kChannel}, kNumberTypeFloat32, "graph_input_1"… in InitGraph() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/custom/aot_test_files/
Dadd_with_attr.cc22 void launch(float *input_1, float *input_2, float *output, float *workspace, int size) { in launch()
Dkernel.cc62 float *input_1 = static_cast<float *>(params[0]); in CustomKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/
Dtensor_util.cc655 bool IsSameDtype(const Tensor *input_1, const Tensor *input_2) { in IsSameDtype()
667 bool IsSameShape(const Tensor *input_1, const Tensor *input_2) { in IsSameShape()
785 bool IsSameDtype(const Tensor *input_1, const Tensor *input_2) { in IsSameDtype()
794 bool IsSameShape(const Tensor *input_1, const Tensor *input_2) { in IsSameShape()
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_transpose_op.py70 def __init__(self, input_1, input_2, perm_1, perm_2): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dbcast.h39 Tensor *input_1; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/
Daddn_cpu_kernel.cc92 const auto input_1 = reinterpret_cast<T *>(inputs[1]->device_ptr()); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/
Dsequence_addn_cpu_kernel.cc89 auto input_1 = input_0 + elements_num; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/
Dany_type_kernel_actor_test.cc121 DataType input_1 = 2.0; in TEST_F() local

123