Home
last modified time | relevance | path

Searched refs:scratch_size (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/
Dfft_util.cc45 size_t scratch_size = 0; in FftPopulateState() local
47 state->fft_size, 0, nullptr, &scratch_size); in FftPopulateState()
52 state->scratch = malloc(scratch_size); in FftPopulateState()
57 state->scratch_size = scratch_size; in FftPopulateState()
60 state->scratch, &scratch_size); in FftPopulateState()
Dfft_io.c21 fprintf(fp, "static char fft_scratch[%zu];\n", state->scratch_size); in FftWriteMemmapPreamble()
32 fprintf(fp, "%s->scratch_size = %zu;\n", variable, state->scratch_size); in FftWriteMemmap()
Dfft.h36 size_t scratch_size; member
/external/mesa3d/src/amd/common/
Dac_binary.c43 uint32_t scratch_size = 0; in ac_parse_shader_binary_config() local
99 scratch_size = value; in ac_parse_shader_binary_config()
126 conf->scratch_bytes_per_wave = G_00B860_WAVESIZE(scratch_size) * 256 * 4; in ac_parse_shader_binary_config()
/external/harfbuzz_ng/src/
Dhb-graphite2.cc238 unsigned int scratch_size; in _hb_graphite2_shape() local
239 hb_buffer_t::scratch_buffer_t *scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_graphite2_shape()
276 scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_graphite2_shape()
278 DIV_CEIL (sizeof (hb_codepoint_t) * glyph_count, sizeof (*scratch))) > scratch_size) in _hb_graphite2_shape()
286 scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_graphite2_shape()
293 assert (_consumed <= scratch_size); \ in _hb_graphite2_shape()
295 scratch_size -= _consumed; \ in _hb_graphite2_shape()
Dhb-coretext.cc592 unsigned int scratch_size; in _hb_coretext_shape() local
593 hb_buffer_t::scratch_buffer_t *scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_coretext_shape()
599 if (unlikely (_consumed > scratch_size)) \ in _hb_coretext_shape()
605 scratch_size -= _consumed; \ in _hb_coretext_shape()
669 scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_coretext_shape()
673 scratch_size -= old_scratch_used; in _hb_coretext_shape()
961 unsigned int scratch_size_saved = scratch_size; \ in _hb_coretext_shape()
965 scratch_size = scratch_size_saved; \ in _hb_coretext_shape()
Dhb-directwrite.cc562 unsigned int scratch_size; in _hb_directwrite_shape_full() local
563 hb_buffer_t::scratch_buffer_t *scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_directwrite_shape_full()
568 assert (_consumed <= scratch_size); \ in _hb_directwrite_shape_full()
570 scratch_size -= _consumed; \ in _hb_directwrite_shape_full()
692 unsigned int glyphs_size = (scratch_size * sizeof (int) - 2) in _hb_directwrite_shape_full()
Dhb-uniscribe.cc731 unsigned int scratch_size; in _hb_uniscribe_shape()
732 hb_buffer_t::scratch_buffer_t *scratch = buffer->get_scratch_buffer (&scratch_size); in _hb_uniscribe_shape()
738 assert (_consumed <= scratch_size); \ in _hb_uniscribe_shape()
740 scratch_size -= _consumed; \ in _hb_uniscribe_shape()
781 unsigned int glyphs_size = (scratch_size * sizeof (int) - 2) in _hb_uniscribe_shape()
/external/mesa3d/src/gallium/drivers/r600/sfn/
Dsfn_instruction_fetch.cpp196 FetchInstruction::FetchInstruction(GPRVector dst, PValue src, int scratch_size): in FetchInstruction() argument
227 m_array_size = scratch_size - 1;
299 LoadFromScratch::LoadFromScratch(GPRVector dst, PValue src, int scratch_size): in LoadFromScratch() argument
300 FetchInstruction(dst, src, scratch_size) in LoadFromScratch()
Dsfn_instruction_fetch.h79 FetchInstruction(GPRVector dst, PValue src, int scratch_size);
169 LoadFromScratch(GPRVector dst, PValue src, int scratch_size);
Dsfn_nir.h86 int scratch_size; variable
Dsfn_shader_base.h60 r600_shader& sh_info, int scratch_size, enum chip_class _chip_class,
/external/mesa3d/src/compiler/nir/
Dnir_lower_scratch.c139 var->data.location = ALIGN_POT(shader->scratch_size, var_align); in nir_lower_vars_to_scratch()
140 shader->scratch_size = var->data.location + var_size; in nir_lower_vars_to_scratch()
Dnir_clone.c775 ns->scratch_size = s->scratch_size; in nir_shader_clone()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h131 size_t scratch_size; member
Dgpu_conv_runner.cc277 desc.scratch_size); in GetGpuConvConfig()
450 descriptor.scratch_size = cudnn_call->shape().tuple_shapes(1).dimensions(0); in GetGpuConvConfig()
Dgpu_conv_algorithm_picker.cc706 result.set_scratch_bytes(profile_result.scratch_size()); in PickBestAlgorithmNoCacheRocm()
726 options.scratch_size_override = miopen_alg.scratch_size(); in PickBestAlgorithmNoCacheRocm()
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_cp_dma.c259 unsigned scratch_size = SI_CPDMA_ALIGNMENT * 2; in si_cp_dma_realign_engine() local
266 if (!sctx->scratch_buffer || sctx->scratch_buffer->b.b.width0 < scratch_size) { in si_cp_dma_realign_engine()
270 PIPE_USAGE_DEFAULT, scratch_size, 256); in si_cp_dma_realign_engine()
/external/mesa3d/src/amd/compiler/
Daco_instruction_selection_setup.cpp1157 unsigned scratch_size = 0; in setup_isel_context() local
1168 scratch_size = std::max(scratch_size, shaders[i]->scratch_size); in setup_isel_context()
1171 ctx.program->config->scratch_bytes_per_wave = align(scratch_size * ctx.program->wave_size, 1024); in setup_isel_context()
/external/tensorflow/tensorflow/core/kernels/
Dxsmm_conv2d.cc401 const size_t scratch_size = libxsmm_dnn_get_scratch_size( in CallLibxsmmConvGeneric() local
405 libxsmm_aligned_scratch(scratch_size, 2097152 /*alignment*/); in CallLibxsmmConvGeneric()
Dconv_grad_input_ops.cc466 result.set_scratch_bytes(profile_result.scratch_size()); in operator ()()
476 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
Dconv_ops_3d.cc587 result.set_scratch_bytes(profile_result.scratch_size()); in launch()
598 miopen_algorithm.scratch_size()), in launch()
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_nir.h222 unsigned scratch_size; member
Dlp_bld_nir_soa.c1850 unsigned scratch_size) in get_scratch_thread_offsets() argument
1860 elems[i] = LLVMConstInt(elem_type, scratch_size * i, 0); in get_scratch_thread_offsets()
1876 …MValueRef thread_offsets = get_scratch_thread_offsets(gallivm, uint_bld->type, bld->scratch_size);; in emit_load_scratch()
1938 …MValueRef thread_offsets = get_scratch_thread_offsets(gallivm, uint_bld->type, bld->scratch_size);; in emit_store_scratch()
2123 if (shader->scratch_size) { in lp_build_nir_soa()
2126 … lp_build_const_int32(gallivm, shader->scratch_size * type.length), in lp_build_nir_soa()
2129 bld.scratch_size = shader->scratch_size; in lp_build_nir_soa()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h787 size_t scratch_size() const { return scratch_size_; } in scratch_size() function
818 AlgorithmConfig(AlgorithmDesc algorithm, size_t scratch_size) in AlgorithmConfig() argument
819 : algorithm_(algorithm), scratch_size_(scratch_size) {} in AlgorithmConfig()
830 absl::optional<size_t> scratch_size() const { return scratch_size_; } in scratch_size() function

12