Home
last modified time | relevance | path

Searched refs:input_sizes (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mesa3d/src/compiler/nir/
Dnir_opcodes.py35 def __init__(self, name, output_size, output_type, input_sizes, argument
70 assert isinstance(input_sizes, list)
71 assert isinstance(input_sizes[0], int)
77 assert len(input_sizes) == len(input_types)
79 for size in input_sizes:
86 self.num_inputs = len(input_sizes)
89 self.input_sizes = input_sizes
155 def opcode(name, output_size, output_type, input_sizes, input_types, argument
158 opcodes[name] = Opcode(name, output_size, output_type, input_sizes,
Dnir_lower_alu_to_scalar.c52 nir_op_infos[alu->op].input_sizes[0] > 1; in inst_is_vector_alu()
68 unsigned num_components = nir_op_infos[alu->op].input_sizes[0]; in lower_reduction()
119 unsigned num_components = nir_op_infos[alu->op].input_sizes[0]; in lower_fdot()
335 assert(nir_op_infos[alu->op].input_sizes[i] < 2); in lower_alu_instr_scalar()
336 unsigned src_chan = (nir_op_infos[alu->op].input_sizes[i] == 1 ? in lower_alu_instr_scalar()
Dnir_inline_uniforms.c74 int input_sizes = nir_op_infos[alu->op].input_sizes[i]; in src_only_uses_uniforms() local
76 if (input_sizes == 0) { in src_only_uses_uniforms()
87 for (unsigned j = 0; j < input_sizes; j++) { in src_only_uses_uniforms()
Dnir_search.c273 if (nir_op_infos[instr->op].input_sizes[src] != 0) { in match_value()
274 num_components = nir_op_infos[instr->op].input_sizes[src]; in match_value()
498 if (nir_op_infos[alu->op].input_sizes[i] != 0) in construct_value()
499 num_components = nir_op_infos[alu->op].input_sizes[i]; in construct_value()
Dnir_lower_regs_to_ssa.c152 if (nir_op_infos[alu->op].input_sizes[i] != 0) in rewrite_alu_instr()
Dnir_lower_vec_to_movs.c182 if (nir_op_infos[src_alu->op].input_sizes[j] != 0) in try_coalesce()
Dnir.h1402 uint8_t input_sizes[NIR_ALU_MAX_INPUTS]; member
1470 if (nir_op_infos[instr->op].input_sizes[src] > 0) in nir_alu_instr_channel_used()
1471 return channel < nir_op_infos[instr->op].input_sizes[src]; in nir_alu_instr_channel_used()
1495 if (nir_op_infos[instr->op].input_sizes[src] > 0) in nir_ssa_alu_instr_src_components()
1496 return nir_op_infos[instr->op].input_sizes[src]; in nir_ssa_alu_instr_src_components()
2793 if (nir_op_infos[alu->op].input_sizes[alu_src_idx] == 0) { in nir_ssa_scalar_chase_alu_src()
2803 assert(nir_op_infos[alu->op].input_sizes[alu_src_idx] == 1); in nir_ssa_scalar_chase_alu_src()
Dnir_opt_vectorize.c154 if (nir_op_infos[alu->op].input_sizes[i] != 0) in instr_can_rewrite()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/
Dapply_momentum_ps_kernel.cc27 const std::vector<size_t> &ApplyMomentumPSKernel::input_sizes() const { return GetInputSizeList(); } in input_sizes() function in mindspore::kernel::ps::ApplyMomentumPSKernel
Dapply_momentum_ps_kernel.h37 const std::vector<size_t> &input_sizes() const override;
Dsparse_apply_adam_ps_kernel.h41 const std::vector<size_t> &input_sizes() const override;
Dsparse_apply_lazy_adam_ps_kernel.h40 const std::vector<size_t> &input_sizes() const override;
Dembedding_look_up_ps_kernel.h41 const std::vector<size_t> &input_sizes() const override;
Dsparse_apply_ftrl_ps_kernel.h42 const std::vector<size_t> &input_sizes() const override;
Dpserver_kernel.h43 virtual const std::vector<size_t> &input_sizes() const = 0;
Dembedding_look_up_ps_kernel.cc107 const std::vector<size_t> &EmbeddingLookUpPSKernel::input_sizes() const { return input_shape_; } in input_sizes() function in mindspore::kernel::ps::EmbeddingLookUpPSKernel
Dsparse_apply_adam_ps_kernel.cc111 const std::vector<size_t> &SparseApplyAdamPSKernel::input_sizes() const { return GetInputSizeList()… in input_sizes() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
Dsparse_apply_lazy_adam_ps_kernel.cc113 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::input_sizes() const { return GetInputSizeLi… in input_sizes() function in mindspore::kernel::ps::SparseApplyLazyAdamPSKernel
Dsparse_apply_ftrl_ps_kernel.cc122 const std::vector<size_t> &SparseApplyFtrlPSKernel::input_sizes() const { return GetInputSizeList()… in input_sizes() function in mindspore::kernel::ps::SparseApplyFtrlPSKernel
/third_party/ffmpeg/tests/checkasm/
Dsw_scale.c72 static const int input_sizes[INPUT_SIZES] = {8, 24, 128, 144, 256, 512}; in check_yuv2yuvX() local
98 dstW = input_sizes[isi]; in check_yuv2yuvX()
/third_party/mesa3d/docs/_exts/
Dnir.py91 for i, t, s in zip(range(100), op.input_types, op.input_sizes):
/third_party/mindspore/mindspore/ccsrc/ps/
Dparameter_server.cc227 const std::vector<size_t> &input_shapes = lookup->input_sizes(); in InitEmbeddingTable()
552 const std::vector<size_t> &input_shapes = lookup->input_sizes(); in SyncEmbeddingTables()
/third_party/mesa3d/src/intel/compiler/
Dbrw_vec4_nir.cpp819 brw_swizzle_for_size(nir_op_infos[cmp_instr->op].input_sizes[0]); in optimize_predicate()
1506 brw_swizzle_for_size(nir_op_infos[instr->op].input_sizes[0]); in nir_emit_alu()
1525 brw_swizzle_for_size(nir_op_infos[instr->op].input_sizes[0]); in nir_emit_alu()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc1156 auto input_sizes = to_kernel_mod->GetInputSizeList(); in LinkDataArrowForCopyActor() local
1157 if (to_input_index >= input_sizes.size()) { in LinkDataArrowForCopyActor()
1158 …OG(EXCEPTION) << "To input index(" << to_input_index << ") is out of size: " << input_sizes.size(); in LinkDataArrowForCopyActor()
1161 …nullptr, input_sizes[to_input_index], from_device_tensor->format(), from_device_tensor->type_id()); in LinkDataArrowForCopyActor()
/third_party/mesa3d/src/broadcom/compiler/
Dv3d_compiler.h921 uint8_t input_sizes[V3D_MAX_GS_INPUTS / 4]; member

12