Home
last modified time | relevance | path

Searched refs:tmp_output (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Drandom_choice_with_mask_cpu_kernel.cc23 int32_t input_total_count, const int *tmp_output, int *output) { in ParseOutputCoordinate() argument
27 int32_t tmp_output_number = tmp_output[i]; in ParseOutputCoordinate()
133 auto *tmp_output = reinterpret_cast<int *>(workspace[1]->addr); in Launch() local
149 (void)memset_s(tmp_output, IntToSize(output_length), 0X00, IntToSize(output_length)); in Launch()
157 tmp_output[i] = input_dim[mean]; in Launch()
163 tmp_output[non_zero_num + index] = 0; in Launch()
176 …ParseOutputCoordinate(dims_, output_length, input_dim_size, input_total_count, tmp_output, output); in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc743 auto [tmp_output, tmp_output_index] = output_index[i]; in GenParallelJson()
745 [&tmp_output](const AnfNodePtr &in) { return tmp_output == in; }); in GenParallelJson()
747 auto tcnode = tmp_output->cast<CNodePtr>(); in GenParallelJson()
758 …GetTensorName(node_json_map.at(tmp_output), kJsonKeyOutputDesc, std::make_pair(0, tmp_output_index… in GenParallelJson()
804 auto tmp_output = output_index[i]; in CreateOutputsJson() local
808 if (tmp_output.first == input_list[input_i]) { in CreateOutputsJson()
815 auto type_id = this->GetOutputDataType(tmp_output.first, tmp_output.second); in CreateOutputsJson()
818 …GetTensorName(node_json_map.at(tmp_output.first), kJsonKeyOutputDesc, std::make_pair(0, tmp_output… in CreateOutputsJson()
820 output_desc_json[kJsonKeyFormat] = this->GetOutputFormat(tmp_output.first, tmp_output.second); in CreateOutputsJson()
821 auto output_shape = this->GetOutputShape(tmp_output.first, tmp_output.second); in CreateOutputsJson()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Ddeconv_int8_tests.cc258 int32_t tmp_output[12 * 24] = {0}; in TEST_F() local
334 MatMulInt8_16x4(packed_a, packed_b, tmp_output, 12, 24, 16, input_sum, weight_sum); in TEST_F()
335 ASSERT_EQ(0, CompareOutputData(tmp_output, correct_tmp_output, 12 * 3 * 8, 0)); in TEST_F()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc513 std::vector<AnfNodePtr> tmp_output{}; in IdentifyBodySubgraphOutput() local
514 tmp_output.resize(input_enter_nodes_.size()); in IdentifyBodySubgraphOutput()
526 tmp_output[pos] = node_pair.second; in IdentifyBodySubgraphOutput()
547 if (tmp_output.size() == 1) { in IdentifyBodySubgraphOutput()
548 return_cnode->add_input(tmp_output[0]); in IdentifyBodySubgraphOutput()
550 std::vector<AnfNodePtr> make_tuple_inputs = tmp_output; in IdentifyBodySubgraphOutput()
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_gs.h64 float (*tmp_output)[4]; member
Ddraw_gs.c429 &shader->stream[i].tmp_output); in gs_flush()
619 shader->stream[i].tmp_output = (float (*)[4])output_verts[i].verts->data; in draw_geometry_shader_run()
/third_party/ffmpeg/libavcodec/
Dac3dec.c700 FFTSample *x = s->tmp_output + 128; in do_imdct()
703 s->imdct_256.imdct_half(&s->imdct_256, s->tmp_output, x); in do_imdct()
706 s->tmp_output, s->window, 128, 8); in do_imdct()
709 s->tmp_output, s->window, 128); in do_imdct()
715 s->imdct_512.imdct_half(&s->imdct_512, s->tmp_output, s->transform_coeffs[ch]); in do_imdct()
718 s->tmp_output, s->window, 128, 8); in do_imdct()
721 s->tmp_output, s->window, 128); in do_imdct()
723 memcpy(s->delay[ch - 1 + offset], s->tmp_output + 128, 128 * sizeof(FFTSample)); in do_imdct()
Dac3dec.h249 …DECLARE_ALIGNED(32, INTFLOAT, tmp_output)[AC3_BLOCK_SIZE]; ///< temporary…