Searched refs:max_scratch_size (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | matmul_util.cc | 111 static const int64_t max_scratch_size = in GetPlanAndAlgorithms() local
184 se::cuda::BlasLt::MatmulPreference::Create(max_scratch_size)); in GetPlanAndAlgorithms()
|
| H A D | matmul_op_impl.h | 403 static const int64_t max_scratch_size =
457 BlasScratchAllocator scratch_allocator(context, max_scratch_size);
491 BlasScratchAllocator scratch_allocator(context, max_scratch_size);
519 BlasScratchAllocator scratch_allocator(context, max_scratch_size);
|
| /aosp_15_r20/external/mesa3d/src/asahi/vulkan/ |
| H A D | hk_cmd_buffer.c | 578 uint32_t max_scratch_size = in hk_reserve_scratch() local
581 if (max_scratch_size == 0) in hk_reserve_scratch()
587 hk_device_alloc_scratch(dev, s->b.info.stage, max_scratch_size); in hk_reserve_scratch()
|
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_lower_shader_calls.c | 560 unsigned max_scratch_size = shader->scratch_size; in spill_ssa_defs_and_lower_shader_calls() local
667 max_scratch_size = MAX2(max_scratch_size, offset); in spill_ssa_defs_and_lower_shader_calls()
707 shader->scratch_size = max_scratch_size; in spill_ssa_defs_and_lower_shader_calls()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/asahi/ |
| H A D | agx_state.c | 2964 uint32_t max_scratch_size = in agx_build_pipeline() local
2967 if (max_scratch_size > 0) { in agx_build_pipeline()
2972 agx_scratch_alloc(&ctx->scratch_fs, max_scratch_size, max_subgroups); in agx_build_pipeline()
2978 agx_scratch_alloc(&ctx->scratch_vs, max_scratch_size, max_subgroups); in agx_build_pipeline()
2984 agx_scratch_alloc(&ctx->scratch_cs, max_scratch_size, max_subgroups); in agx_build_pipeline()
|
| /aosp_15_r20/external/mesa3d/src/intel/compiler/elk/ |
| H A D | elk_fs.cpp | 5882 ASSERTED unsigned max_scratch_size = 2 * 1024 * 1024; in allocate_registers() local
5905 max_scratch_size = 12 * 1024; in allocate_registers()
5919 assert(prog_data->total_scratch < max_scratch_size); in allocate_registers()
|