Home
last modified time | relevance | path

Searched defs:input_sizes (Results 1 – 11 of 11) sorted by relevance

/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
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
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_adam_ps_kernel.cc111 const std::vector<size_t> &SparseApplyAdamPSKernel::input_sizes() const { return GetInputSizeList()… in input_sizes() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
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
/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
/third_party/mesa3d/src/compiler/nir/
Dnir_inline_uniforms.c74 int input_sizes = nir_op_infos[alu->op].input_sizes[i]; in src_only_uses_uniforms() local
Dnir_opcodes.py35 def __init__(self, name, output_size, output_type, input_sizes, argument
155 def opcode(name, output_size, output_type, input_sizes, input_types, argument
Dnir.h1402 uint8_t input_sizes[NIR_ALU_MAX_INPUTS]; member
/third_party/mesa3d/src/broadcom/compiler/
Dv3d_compiler.h921 uint8_t input_sizes[V3D_MAX_GS_INPUTS / 4]; member
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc1156 auto input_sizes = to_kernel_mod->GetInputSizeList(); in LinkDataArrowForCopyActor() local