| /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/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | format_transpose_infer.c | 43 int input_b = GetBatch(input); in FormatTransposeInferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/export_and_load/ |
| D | test_get_and_init_graph_cell_parameters.py | 50 input_b = Tensor(np_b) variable
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/mindir/ |
| D | test_init_graph_cell_parameters_with_illegal_data.py | 49 input_b = Tensor(np_b) variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/ |
| D | arithmetic_coreml.cc | 22 auto input_b = in_tensors_.at(1); in IsSupport() local
|
| D | matmul_coreml.cc | 71 auto input_b = in_tensors_.at(SECOND_INPUT); in ConstMatMulWithTransB() local
|
| /third_party/vixl/examples/aarch64/ |
| D | debugging.cc | 75 int32_t input_b = 2; in main() local
|
| D | simulated-runtime-calls.cc | 109 int32_t input_b = 2; 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 | multilabel_margin_loss_grad_impl.cu | 35 const T *input_b = input + bid * class_num; in MultilabelMarginLossGradKernel() local
99 const half *input_b = input + bid * class_num; in MultilabelMarginLossGradKernel() local
|
| 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()
|
| D | multilabel_margin_loss_impl.cu | 56 const T *input_b = input + bid * class_num; in MultilabelMarginLossKernel() local
130 const half *input_b = input + bid * class_num; in MultilabelMarginLossKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | lu_solve_cpu_kernel.cc | 87 Map<Matrix<T, RowMajor>> input_b(b_value, b_row_, b_col_); in LaunchKernel() local
|
| /third_party/protobuf/third_party/utf8_range/ |
| D | range2-sse.c | 102 const __m128i input_b = _mm_loadu_si128((const __m128i *)(data+16)); in utf8_range2() local
|
| D | range2-neon.c | 63 const uint8x16_t input_b = vld1q_u8(data + 16); in utf8_range2() local
|
| /third_party/grpc/third_party/utf8_range/ |
| D | range2-sse.c | 102 const __m128i input_b = _mm_loadu_si128((const __m128i *)(data+16)); in utf8_range2() local
|
| D | range2-neon.c | 63 const uint8x16_t input_b = vld1q_u8(data + 16); in utf8_range2() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | betainc_gpu_kernel.cc | 77 T *input_b = GetDeviceAddress<T>(inputs, kBIndex); 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/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | betainc.cc | 118 auto input_b = reinterpret_cast<T *>(ctx.Input(1)->GetData()); in BetaincCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/ |
| D | matmul_cpu_kernel_func.cc | 165 const auto input_b = reinterpret_cast<float *>(inputs[1]->device_ptr()); in RunFunc() local
|
| /third_party/skia/third_party/externals/tint/src/resolver/ |
| D | entry_point_validation_test.cc | 316 auto* input_b = Structure( in TEST_F() local
786 auto* input_b = Structure( 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
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | op_node.cc | 271 const auto &input_b = inputs_tm[1]; in CalcByOperator() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 3231 auto input_b = node->input(kTwoNum); // matmul bias in ExportPrimDense() local
4240 auto input_b = node->input(kTwoNum); // conv bias in ExportMergeConv() local
4253 auto input_b = node->input(kTwoNum); // matmul bias in ExportMergeGemm() local
|