Home
last modified time | relevance | path

Searched refs:scratch_memory_size (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2923 size_t scratch_memory_size = *(algorithm_config.scratch_size()); in DoPrepareForConvolution() local
2926 if (scratch_memory_size != 0) { in DoPrepareForConvolution()
2932 auto allocated = scratch_allocator->AllocateBytes(scratch_memory_size); in DoPrepareForConvolution()
2944 "Failed to allocate scratch memory of size: ", scratch_memory_size)); in DoPrepareForConvolution()
3464 size_t scratch_memory_size = 0; in GetMIOpenConvolveAlgorithmsFindMode() local
3469 output_nd.handle(), &scratch_memory_size); in GetMIOpenConvolveAlgorithmsFindMode()
3481 input_nd.handle(), &scratch_memory_size); in GetMIOpenConvolveAlgorithmsFindMode()
3493 filter.handle(), &scratch_memory_size); in GetMIOpenConvolveAlgorithmsFindMode()
3512 if (scratch_memory_size != 0) { in GetMIOpenConvolveAlgorithmsFindMode()
3518 auto allocated = scratch_allocator->AllocateBytes(scratch_memory_size); in GetMIOpenConvolveAlgorithmsFindMode()
[all …]
/aosp_15_r20/external/mesa3d/src/amd/common/
H A Dac_rgp_elf_object_pack.c194 record->shader_data[i].scratch_memory_size); in ac_rgp_write_msgpack()
244 record->shader_data[i].scratch_memory_size); in ac_rgp_write_msgpack()
H A Dac_rgp.h38 uint32_t scratch_memory_size; member
/aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/
H A Dsi_sqtt.c790 record->shader_data[i].scratch_memory_size = shader->config.scratch_bytes_per_wave; in si_sqtt_add_code_object()
/aosp_15_r20/external/mesa3d/src/amd/vulkan/layers/
H A Dradv_sqtt_layer.c1439 shader_data->scratch_memory_size = shader->config.scratch_bytes_per_wave; in radv_fill_code_object_record()