Home
last modified time | relevance | path

Searched refs:main_fn (Results 1 – 13 of 13) sorted by relevance

/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_shader_llvm.c184 ctx->main_fn = ac_build_main(&ctx->args, &ctx->ac, call_conv, name, ret_type, ctx->ac.module); in si_llvm_create_func()
188 ac_llvm_add_target_dep_function_attr(ctx->main_fn, "amdgpu-32bit-address-high-bits", in si_llvm_create_func()
192 ac_llvm_set_workgroup_size(ctx->main_fn, max_workgroup_size); in si_llvm_create_func()
259 ptr[0] = LLVMGetParam(ctx->main_fn, (merged_shader ? 8 : 0) + SI_SGPR_RW_BUFFERS); in si_prolog_get_rw_buffers()
386 LLVMValueRef main_fn = ctx->main_fn; in si_nir_build_llvm() local
395 values[0] = mask & 0x1 ? LLVMGetParam(main_fn, offset++) : undef; in si_nir_build_llvm()
396 values[1] = mask & 0x2 ? LLVMGetParam(main_fn, offset++) : undef; in si_nir_build_llvm()
397 values[2] = mask & 0x4 ? LLVMGetParam(main_fn, offset++) : undef; in si_nir_build_llvm()
398 values[3] = mask & 0x8 ? LLVMGetParam(main_fn, offset++) : undef; in si_nir_build_llvm()
404 values[0] = mask & 0x1 ? LLVMGetParam(main_fn, offset++) : undef; in si_nir_build_llvm()
[all …]
Dsi_shader_llvm_ps.c211 LLVMValueRef alpha_ref = LLVMGetParam(ctx->main_fn, SI_PARAM_ALPHA_REF); in si_alpha_test()
228 coverage = LLVMGetParam(ctx->main_fn, samplemask_param); in si_scale_alpha_by_sample_mask()
541 builder, ret, ac_to_integer(&ctx->ac, LLVMGetParam(ctx->main_fn, SI_PARAM_ALPHA_REF)), in si_llvm_return_fs_outputs()
572 ret = LLVMBuildInsertValue(builder, ret, LLVMGetParam(ctx->main_fn, SI_PARAM_SAMPLE_COVERAGE), in si_llvm_return_fs_outputs()
656 func = ctx->main_fn; in si_llvm_build_ps_prolog()
899 ac_llvm_add_target_dep_function_attr(ctx->main_fn, "InitialPSInputAddr", 0xffffff); in si_llvm_build_ps_epilog()
931 color[i] = LLVMGetParam(ctx->main_fn, vgpr + i / 2); in si_llvm_build_ps_epilog()
938 color[i] = LLVMGetParam(ctx->main_fn, vgpr++); in si_llvm_build_ps_epilog()
949 depth = LLVMGetParam(ctx->main_fn, vgpr++); in si_llvm_build_ps_epilog()
951 stencil = LLVMGetParam(ctx->main_fn, vgpr++); in si_llvm_build_ps_epilog()
[all …]
Dsi_shader_llvm_vs.c58 LLVMValueRef x1y1 = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs); in load_input_vs()
59 LLVMValueRef x2y2 = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 1); in load_input_vs()
71 out[2] = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 2); in load_input_vs()
81 out[i] = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 3 + i); in load_input_vs()
85 LLVMValueRef x1 = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 3); in load_input_vs()
86 LLVMValueRef y1 = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 4); in load_input_vs()
87 LLVMValueRef x2 = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 5); in load_input_vs()
88 LLVMValueRef y2 = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 6); in load_input_vs()
92 out[2] = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 7); in load_input_vs()
93 out[3] = LLVMGetParam(ctx->main_fn, param_vs_blit_inputs + 8); in load_input_vs()
[all …]
Dsi_shader.c737 ctx->main_fn, "InitialPSInputAddr", in si_create_function()
1309 ac_optimize_vs_outputs(&ctx->ac, ctx->main_fn, shader->info.vs_output_param_offset, in si_optimize_vs_outputs()
1687 ngg_cull_main_fn = ctx.main_fn; in si_llvm_compile_shader()
1688 ctx.main_fn = NULL; in si_llvm_compile_shader()
1700 LLVMValueRef main_fn = ctx.main_fn; in si_llvm_compile_shader() local
1709 parts[num_parts++] = ctx.main_fn; in si_llvm_compile_shader()
1721 parts[num_parts++] = ctx.main_fn; in si_llvm_compile_shader()
1724 parts[num_parts++] = main_fn; in si_llvm_compile_shader()
1734 parts[1] = ctx.main_fn; in si_llvm_compile_shader()
1745 parts[2] = ctx.main_fn; in si_llvm_compile_shader()
[all …]
Dsi_shader_internal.h67 LLVMValueRef main_fn; member
Dsi_shader_llvm_gs.c483 end_bb = LLVMAppendBasicBlockInContext(ctx.ac.context, ctx.main_fn, "end"); in si_generate_gs_copy_shader()
598 func = ctx->main_fn; in si_llvm_build_gs_prolog()
Dsi_compute_prim_discard.c304 LLVMValueRef vs = ctx->main_fn; in si_build_prim_discard_compute_shader()
355 ac_llvm_add_target_dep_function_attr(ctx->main_fn, "amdgpu-gds-size", 256); in si_build_prim_discard_compute_shader()
357 ac_llvm_add_target_dep_function_attr(ctx->main_fn, "amdgpu-gds-size", GDS_SIZE_UNORDERED); in si_build_prim_discard_compute_shader()
Dgfx10_shader_ngg.c290 ac_llvm_add_target_dep_function_attr(ctx->main_fn, "amdgpu-gds-size", 256); in build_streamout()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc520 mlir::FuncOp main_fn = module_op.lookupSymbol<mlir::FuncOp>("main"); in RewriteWithArgs() local
524 auto builder = mlir::OpBuilder(main_fn.getBody()); in RewriteWithArgs()
528 mlir::BlockArgument mlir_arg = main_fn.getArgument(idx); in RewriteWithArgs()
573 updated_argument_types.reserve(main_fn.getNumArguments()); in RewriteWithArgs()
574 for (mlir::BlockArgument& arg : main_fn.getArguments()) in RewriteWithArgs()
577 main_fn.setType(mlir::FunctionType::get(main_fn.getContext(), in RewriteWithArgs()
579 main_fn.getType().getResults())); in RewriteWithArgs()
582 for (int idx : llvm::reverse(args_to_erase)) main_fn.eraseArgument(idx); in RewriteWithArgs()
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_run.py254 self.main_fn = fn
323 self.main_result = self.main_fn(*self.main_args, **self.main_kwargs)
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc311 FuncOp main_fn = module.lookupSymbol<FuncOp>("main"); in IsValidTFLiteMlirModule() local
312 if (!main_fn) { in IsValidTFLiteMlirModule()
1733 FuncOp main_fn = module_.lookupSymbol<FuncOp>("main"); in TranslateInternal() local
1734 subgraph_index_map_[main_fn.getName().str()] = subgraph_idx++; in TranslateInternal()
1735 named_regions.emplace_back("main", &main_fn.getBody()); in TranslateInternal()
1738 if (fn != main_fn) { in TranslateInternal()
1831 main_fn, saved_model_tags_.empty() ? "" : *saved_model_tags_.begin()); in TranslateInternal()
/external/mesa3d/src/amd/llvm/
Dac_llvm_build.h448 void ac_optimize_vs_outputs(struct ac_llvm_context *ac, LLVMValueRef main_fn,
Dac_llvm_build.c2831 void ac_optimize_vs_outputs(struct ac_llvm_context *ctx, LLVMValueRef main_fn, in ac_optimize_vs_outputs() argument
2842 bb = LLVMGetFirstBasicBlock(main_fn); in ac_optimize_vs_outputs()
3090 LLVMValueRef main_fn = LLVMGetBasicBlockParent(LLVMGetInsertBlock(ctx->builder)); in append_basic_block() local
3091 return LLVMAppendBasicBlockInContext(ctx->context, main_fn, name); in append_basic_block()