| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_dynamic_rank/ |
| D | test_biasadd_dynamic_rank.py | 34 def construct(self, input_a, input_b, indices): argument
46 def forward(self, input_a, input_b, indices): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/export_and_load/ |
| D | test_get_and_init_graph_cell_parameters.py | 49 input_a = Tensor(np_a) variable
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/mindir/ |
| D | test_init_graph_cell_parameters_with_illegal_data.py | 48 input_a = Tensor(np_a) variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/ |
| D | arithmetic_coreml.cc | 21 auto input_a = in_tensors_.at(0); in IsSupport() local
|
| /third_party/vixl/examples/aarch64/ |
| D | debugging.cc | 74 int32_t input_a = 1; in main() local
|
| D | simulated-runtime-calls.cc | 108 int32_t input_a = 1; in main() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | betainc_impl.cu | 25 cudaError_t CalBetainc(const size_t size, T *input_a, T *input_b, T *input_x, T *output, const uint… in CalBetainc()
|
| D | einsum_impl.cu | 248 __global__ void Dot(const size_t size, T *input_a, const T *input_b, double *output, T *out_t) { in Dot()
274 cudaError_t CalDot(const size_t size, T *input_a, const T *input_b, T *output, cudaStream_t cuda_st… in CalDot()
293 __global__ void DotGrad(const size_t size, const T dout, T *mid_res, T *input_b, T *input_a) { in DotGrad()
300 cudaError_t CalDotGrad(const size_t size, const T dout, T *mid_res, T *input_b, T *input_a, cudaStr… in CalDotGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | svd_cpu_kernel.cc | 86 auto *input_a = reinterpret_cast<T *>(inputs[kIndex0]->device_ptr()); in LaunchKernelFloat() local
146 auto *input_a = reinterpret_cast<T *>(inputs[kIndex0]->device_ptr()); in LaunchKernelComplex() local
|
| D | polygamma_cpu_kernel.cc | 206 auto input_a = reinterpret_cast<T1 *>(inputs[0]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | lu_solve_cpu_kernel.cc | 84 Map<Matrix<T, RowMajor>> input_a(a_value, a_row_, a_col_); in LaunchKernel() local
|
| D | qr_scipy_cpu_kernel.cc | 74 …Eigen::Map<Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>> input_a(a_value, a_r… in LaunchKernel() local
|
| /third_party/protobuf/third_party/utf8_range/ |
| D | range2-sse.c | 61 const __m128i input_a = _mm_loadu_si128((const __m128i *)data); in utf8_range2() local
|
| D | range2-neon.c | 62 const uint8x16_t input_a = vld1q_u8(data); in utf8_range2() local
|
| /third_party/grpc/third_party/utf8_range/ |
| D | range2-sse.c | 61 const __m128i input_a = _mm_loadu_si128((const __m128i *)data); in utf8_range2() local
|
| D | range2-neon.c | 62 const uint8x16_t input_a = vld1q_u8(data); in utf8_range2() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_auto_parallel_rec_unknownop.py | 45 def construct(self, input_a, label): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | betainc_gpu_kernel.cc | 76 T *input_a = GetDeviceAddress<T>(inputs, kAIndex); in LaunchKernel() local
|
| D | einsum_helper.h | 199 …void Bmm(T *input_a, T *input_b, T *output, const std::vector<size_t> &a_shape, const std::vector<… in Bmm()
242 …void Dot(T *input_a, T *input_b, T *output, const std::vector<size_t> &a_shape, const std::vector<… in Dot()
252 void DotGrad(T *output, T *mid_res, T *input_b, T *input_a, const std::vector<size_t> &out_shape, in DotGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | polygamma.cc | 47 int64_t input_a; in PolygammaInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | betainc.cc | 117 auto input_a = reinterpret_cast<T *>(ctx.Input(0)->GetData()); in BetaincCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/ |
| D | matmul_cpu_kernel_func.cc | 164 const auto input_a = reinterpret_cast<float *>(inputs[0]->device_ptr()); in RunFunc() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | matmul_tensorrt.cc | 179 nvinfer1::ITensor *input_a = input(ctx, 0).trt_tensor_; in AddAsFullConnect() local
|
| /third_party/skia/third_party/externals/tint/src/resolver/ |
| D | entry_point_validation_test.cc | 314 auto* input_a = Structure( in TEST_F() local
785 auto* input_a = Structure("InputA", {Member("a", ty.f32(), {Location(1)})}); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/ |
| D | grad_math_ops.py | 183 def bprop(input_a, input_b, out, dout): argument
|