Home
last modified time | relevance | path

Searched refs:input_info (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/ops/_op_impl/_custom_op/
Dbatch_matmul_impl.py65 def _inner_matmul_new(tik_instance, dtype, input_info, res, res_index): argument
67 input1, input1_index, input2, input2_index = input_info
98 def _inner_matmul_new_1_64_32_64(tik_instance, dtype, input_info, res, res_index): argument
100 input1, input1_index, input2, input2_index = input_info
178 input_info = input1, input1_index, input2, input2_index
179 _inner_matmul_new(tik_instance, dtype, input_info, res, res_index)
184 input_info = input1, input1_index, input2, input2_index
185 _inner_matmul_new(tik_instance, dtype, input_info, res, res_index)
224 input_info = input1, input1_index, input2, input2_index
225 _inner_matmul_new(tik_instance, dtype, input_info, res, res_index)
[all …]
Dfused_abs_max1_impl.py57 def _error_feedback(input_info): argument
72 if input_info not in support_shape:
73 raise RuntimeError("input_shape %s is not supported" % str(input_info))
505 input_info = (tuple(input_x_shape), dtype)
509 _error_feedback(input_info)
511 if input_info == ((1, 128, 128), "float32"):
513 elif input_info == ((2, 128, 128), "float32"):
515 elif input_info == ((4, 128, 128), "float32"):
517 elif input_info == ((8, 128, 128), "float32"):
519 elif input_info == ((16, 128, 128), "float32"):
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dstrided_slice.cc108 auto input_info = GpuTensorInfo(in_tensors_.front()); in InitConstArgs() local
110 input_shape_ = {static_cast<cl_int>(input_info.N), static_cast<cl_int>(input_info.H), in InitConstArgs()
111 static_cast<cl_int>(input_info.W), static_cast<cl_int>(input_info.C)}; in InitConstArgs()
114 io_slices_ = {static_cast<cl_int>(input_info.Slice), static_cast<cl_int>(output_info.Slice)}; in InitConstArgs()
121 Broadcast2GpuShape(begin_.s, begin, input_info.NDim, 0); in InitConstArgs()
122 Broadcast2GpuShape(size_.s, size, input_info.NDim, -1); in InitConstArgs()
148 Broadcast2GpuShape(begin_.s, begin, input_info.NDim, 0); in InitConstArgs()
149 Broadcast2GpuShape(end_.s, end, input_info.NDim); in InitConstArgs()
150 Broadcast2GpuShape(stride_.s, stride, input_info.NDim, 1); in InitConstArgs()
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_tess.h66 const struct tgsi_shader_info *input_info; member
93 const struct tgsi_shader_info *input_info; member
109 const struct tgsi_shader_info *input_info,
119 const struct tgsi_shader_info *input_info,
Ddraw_tess.c39 const struct tgsi_shader_info *input_info) in draw_tes_get_input_index() argument
42 const ubyte *input_semantic_names = input_info->output_semantic_name; in draw_tes_get_input_index()
43 const ubyte *input_semantic_indices = input_info->output_semantic_index; in draw_tes_get_input_index()
80 shader->input_info); in llvm_fetch_tcs_input()
161 const struct tgsi_shader_info *input_info, in draw_tess_ctrl_shader_run() argument
177 shader->input_info = input_info; in draw_tess_ctrl_shader_run()
245 shader->input_info); in llvm_fetch_tes_input()
278 TGSI_SEMANTIC_TESSOUTER, 0, shader->input_info); in llvm_fetch_tess_factors()
280 TGSI_SEMANTIC_TESSINNER, 0, shader->input_info); in llvm_fetch_tess_factors()
326 const struct tgsi_shader_info *input_info, in draw_tess_eval_shader_run() argument
[all …]
Ddraw_gs.h94 const struct tgsi_shader_info *input_info; member
140 const struct tgsi_shader_info *input_info,
Ddraw_gs.c50 const struct tgsi_shader_info *input_info) in draw_gs_get_input_index() argument
53 const ubyte *input_semantic_names = input_info->output_semantic_name; in draw_gs_get_input_index()
54 const ubyte *input_semantic_indices = input_info->output_semantic_index; in draw_gs_get_input_index()
162 shader->input_info); in tgsi_fetch_gs_input()
263 shader->input_info); in llvm_fetch_gs_input()
556 const struct tgsi_shader_info *input_info, in draw_geometry_shader_run() argument
625 shader->input_info = input_info; in draw_geometry_shader_run()
/third_party/mindspore/mindspore/lite/test/st/scripts/nnie/
Drun_benchmark_nnie.sh15 input_info=`echo ${nnie_line_info}|awk -F ' ' '{print $3}'`
23 if [[ $input_info != 1 ]]; then
24 input_num=`echo ${input_info} | awk -F ':' '{print $1}'`
25 input_seq=`echo ${input_info} | awk -F ':' '{print $2}'`
/third_party/gstreamer/gstplugins_bad/sys/applemedia/
Dvideotexturecache-gl.m193 switch (GST_VIDEO_INFO_FORMAT (&cache->input_info)) {
197 GST_VIDEO_INFO_WIDTH (&cache->input_info),
198 GST_VIDEO_INFO_HEIGHT (&cache->input_info),
216 GST_VIDEO_INFO_COMP_WIDTH (&cache->input_info, plane),
217 GST_VIDEO_INFO_COMP_HEIGHT (&cache->input_info, plane),
236 gl_target, texformat, CVOpenGLESTextureGetName (texture), &cache->input_info,
Dvideotexturecache.m53 gst_video_info_init (&cache->input_info);
80 gst_video_info_from_caps (&cache->input_info, in_caps);
Dvideotexturecache.h40 GstVideoInfo input_info; member
/third_party/mbedtls/tests/scripts/
Dgenerate_test_code.py1024 def generate_code(**input_info): argument
1040 funcs_file = input_info['funcs_file']
1041 data_file = input_info['data_file']
1042 template_file = input_info['template_file']
1043 platform_file = input_info['platform_file']
1044 helpers_file = input_info['helpers_file']
1045 suites_dir = input_info['suites_dir']
1046 c_file = input_info['c_file']
1047 out_data_file = input_info['out_data_file']
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dcommon.cc65 for (auto input_info : input_infos) { in TestMain() local
66 auto &shape = std::get<0>(input_info); in TestMain()
67 auto category = std::get<2>(input_info); in TestMain()
68 auto data_type = std::get<3>(input_info); in TestMain()
235 for (auto input_info : input_infos) { in TestMain() local
236 auto &shape = std::get<0>(input_info); in TestMain()
237 auto category = std::get<2>(input_info); in TestMain()
238 auto data_type = std::get<3>(input_info); in TestMain()
/third_party/mindspore/mindspore/lite/test/st/scripts/
Dbase_functions.sh142 input_info=`echo ${model_info} | awk -F ';' '{print $2}'`
146 input_num=`echo ${input_info} | sed 's/:/;/' | awk -F ';' '{print $1}'`
147 input_names=`echo ${input_info} | sed 's/:/;/' | awk -F ';' '{print $2}'`
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Doperator_costmodel_test.cc174 TensorInfo input_info(input_layout, input_shape, input_slice_shape); in TEST_F() local
178 inputs.push_back(input_info); in TEST_F()
/third_party/gstreamer/gstplugins_bad/sys/nvcodec/
Dgstnvbaseenc.h150 GstVideoInfo input_info; /* buffer configuration for buffers sent to NVENC */ member
Dgstnvbaseenc.c1250 if (nvenc->input_info.fps_n > 0 && nvenc->input_info.fps_d > 0) { in gst_nv_base_enc_bitstream_thread()
1252 gst_util_uint64_scale (GST_SECOND, nvenc->input_info.fps_d, in gst_nv_base_enc_bitstream_thread()
1253 nvenc->input_info.fps_n); in gst_nv_base_enc_bitstream_thread()
1851 nvenc->input_info = *info; in gst_nv_base_enc_set_format()
2190 dest_stride = _get_cuda_device_stride (&nvenc->input_info, in _map_gl_input_buffer()
2217 data_pointer += dest_stride * _get_plane_height (&nvenc->input_info, i); in _map_gl_input_buffer()
2247 guint dest_stride = _get_cuda_device_stride (&nvenc->input_info, i, in gst_nv_base_enc_upload_frame()
2274 dst += dest_stride * _get_plane_height (&nvenc->input_info, i); in gst_nv_base_enc_upload_frame()
/third_party/gstreamer/gstplugins_good/gst/interleave/
Dinterleave.c1216 GstMapInfo input_info; in gst_interleave_collected() local
1241 gst_buffer_map (inbuf, &input_info, GST_MAP_READ); in gst_interleave_collected()
1242 self->func (outdata, input_info.data, self->channels, nsamples); in gst_interleave_collected()
1243 gst_buffer_unmap (inbuf, &input_info); in gst_interleave_collected()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dauto_monad.cc776 auto input_info = TraceEffectInfo(arg); in TraceEffectInfo() local
777 info.Merge(input_info); in TraceEffectInfo()