Home
last modified time | relevance | path

Searched refs:pos_index (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/
Dbbox_assign_sample.py131pos_index, valid_pos_index = self.random_choice_with_mask_pos(self.greater(assigned_gt_inds5, 0))
136pos_index = pos_index * self.reshape(self.cast(valid_pos_index, mstype.int32), (self.num_expected_…
138 pos_assigned_gt_index = self.gatherND(assigned_gt_inds5, pos_index) - self.assigned_pos_ones
149 pos_bboxes_ = self.gatherND(bboxes, pos_index)
157 bbox_targets_total = self.scatterNd(pos_index, pos_bbox_targets_, (self.num_bboxes, 4))
158 bbox_weights_total = self.scatterNd(pos_index, valid_pos_index, (self.num_bboxes,))
159 labels_total = self.scatterNd(pos_index, pos_gt_labels, (self.num_bboxes,))
160 total_index = self.concat((pos_index, neg_index))
Dbbox_assign_sample_stage2.py151pos_index, valid_pos_index = self.random_choice_with_mask_pos(self.greater(assigned_gt_inds5, 0))
156pos_index = pos_index * self.reshape(self.cast(valid_pos_index, mstype.int32), (self.num_expected_…
160 pos_index = self.reshape(pos_index, self.reshape_shape_pos)
162 pos_index = pos_index * valid_pos_index
164 pos_assigned_gt_index = self.gatherND(assigned_gt_inds5, pos_index) - self.assigned_pos_ones
181 pos_bboxes_ = self.gatherND(bboxes, pos_index)
/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dutil_layout_gen_test.cc95 TensorMap GenerateTensorMap(const int64_t &map_size, const Shape &pos_index, const Shape &pos_value… in GenerateTensorMap() argument
97 for (size_t i = 0; i < pos_index.size() && i < pos_value.size(); i++) { in GenerateTensorMap()
98 if (pos_index[i] >= map_size) { in GenerateTensorMap()
101 tensor_map[pos_index[i]] = pos_value[i]; in GenerateTensorMap()
122 std::vector<Shape> pos_index; in GenerateValidTensorMap() local
123 pos_index = combine(pos_ind_combine_in, pos_num); in GenerateValidTensorMap()
140 for (size_t j = 0; j < pos_index.size(); j++) { in GenerateValidTensorMap()
142 TensorMap tensor_map = GenerateTensorMap(shape_size, pos_index[j], pos_value[k]); in GenerateValidTensorMap()
Dutil_layout_gen_test.h35 TensorMap GenerateTensorMap(const int64_t &map_size, const Shape &pos_index, const Shape &pos_value…
/third_party/boost/libs/numeric/odeint/examples/
Dmolecular_dynamics_cells.cpp229 index_vector pos_index( m_p.n ); in operator ()() local
239 …boost::make_zip_iterator( boost::make_tuple( x.begin() , pos_index.begin() , pos_hash.begin() ) ) , in operator ()()
240 … boost::make_zip_iterator( boost::make_tuple( x.end() , pos_index.end() , pos_hash.end() ) ) , in operator ()()
263 pos_index.begin() , in operator ()()
270 pos_index.end() , in operator ()()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dmirror_pad_cpu_kernel.cc196 …auto pos_index = (equiv_block_num * old_height + matchval_y_index - paddings[HEIGHT]) * old_width + in LaunchKernel() local
198 outputs_addr[pos] = inputs_addr[pos_index]; in LaunchKernel()
/third_party/python/Lib/
D_pyio.py999 pos_index = pos.__index__
1003 pos = pos_index()
1028 pos_index = pos.__index__
1032 pos = pos_index()
/third_party/flutter/engine/flutter/third_party/txt/src/txt/
Dparagraph_txt.cc898 size_t pos_index = blob_index * 2; in Layout() local
899 blob_buffer.pos[pos_index] = in Layout()
901 blob_buffer.pos[pos_index + 1] = layout.getY(glyph_index); in Layout()
904 glyph_x_offset = blob_buffer.pos[pos_index]; in Layout()
/third_party/mesa3d/src/gallium/drivers/r600/
Dr600_state.c2445 int pos_index = -1, face_index = -1, fixed_pt_position_index = -1; in r600_update_ps_state() local
2460 pos_index = i; in r600_update_ps_state()
2538 if (pos_index != -1) { in r600_update_ps_state()
2540 …S_0286CC_POSITION_CENTROID(rshader->input[pos_index].interpolate_location == TGSI_INTERPOLATE_LOC_… in r600_update_ps_state()
2541 S_0286CC_POSITION_ADDR(rshader->input[pos_index].gpr) | in r600_update_ps_state()
2543 …S_0286CC_POSITION_SAMPLE(rshader->input[pos_index].interpolate_location == TGSI_INTERPOLATE_LOC_SA… in r600_update_ps_state()
Devergreen_state.c3326 int pos_index = -1, face_index = -1, fixed_pt_position_index = -1; in evergreen_update_ps_state() local
3352 pos_index = i; in evergreen_update_ps_state()
3478 if (pos_index != -1) { in evergreen_update_ps_state()
3480 …S_0286CC_POSITION_CENTROID(rshader->input[pos_index].interpolate_location == TGSI_INTERPOLATE_LOC_… in evergreen_update_ps_state()
3481 S_0286CC_POSITION_ADDR(rshader->input[pos_index].gpr); in evergreen_update_ps_state()
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dgfx10_shader_ngg.c832 unsigned pos_index = 0; in gfx10_emit_ngg_culling_epilogue() local
850 pos_index = i; in gfx10_emit_ngg_culling_epilogue()
1085 … builder, ac_to_integer(&ctx->ac, LLVMBuildLoad(builder, addrs[4 * pos_index + chan], "")), in gfx10_emit_ngg_culling_epilogue()