Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc200 PrimitiveType prim_type, llvm::Value* value) { in EmitErfcInv() argument
201 return EmitLibdeviceMathCall("__nv_erfcinv", {value}, {prim_type}, prim_type); in EmitErfcInv()
204 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitLog(PrimitiveType prim_type, in EmitLog() argument
206 return EmitLibdeviceMathCall("__nv_log", {value}, {prim_type}, prim_type); in EmitLog()
209 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitLog1p(PrimitiveType prim_type, in EmitLog1p() argument
211 return EmitLibdeviceMathCall("__nv_log1p", {value}, {prim_type}, prim_type); in EmitLog1p()
214 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitSin(PrimitiveType prim_type, in EmitSin() argument
216 return EmitLibdeviceMathCall("__nv_sin", {value}, {prim_type}, prim_type); in EmitSin()
219 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitCos(PrimitiveType prim_type, in EmitCos() argument
221 return EmitLibdeviceMathCall("__nv_cos", {value}, {prim_type}, prim_type); in EmitCos()
[all …]
Delemental_ir_emitter.h58 StatusOr<llvm::Value*> EmitErfcInv(PrimitiveType prim_type,
61 StatusOr<llvm::Value*> EmitLog(PrimitiveType prim_type,
64 StatusOr<llvm::Value*> EmitLog1p(PrimitiveType prim_type,
67 StatusOr<llvm::Value*> EmitSin(PrimitiveType prim_type,
70 StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
73 StatusOr<llvm::Value*> EmitExp(PrimitiveType prim_type,
76 StatusOr<llvm::Value*> EmitExpm1(PrimitiveType prim_type,
79 StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
82 StatusOr<llvm::Value*> EmitRsqrt(PrimitiveType prim_type,
85 StatusOr<llvm::Value*> EmitPow(PrimitiveType prim_type, llvm::Value* lhs,
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h110 virtual StatusOr<llvm::Value*> EmitErfInv(PrimitiveType prim_type,
113 virtual StatusOr<llvm::Value*> EmitErfcInv(PrimitiveType prim_type,
116 virtual StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type,
119 virtual StatusOr<llvm::Value*> EmitLog(PrimitiveType prim_type,
122 virtual StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
125 virtual StatusOr<llvm::Value*> EmitRsqrt(PrimitiveType prim_type,
128 virtual StatusOr<llvm::Value*> EmitLog1p(PrimitiveType prim_type,
131 virtual StatusOr<llvm::Value*> EmitSin(PrimitiveType prim_type,
134 virtual StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
137 virtual StatusOr<llvm::Value*> EmitExp(PrimitiveType prim_type,
[all …]
Delemental_ir_emitter.cc892 StatusOr<llvm::Value*> ElementalIrEmitter::EmitErfInv(PrimitiveType prim_type, in EmitErfInv() argument
894 if (prim_type != F16 && prim_type != F32 && prim_type != F64) { in EmitErfInv()
901 if (prim_type == F16) { in EmitErfInv()
939 if (prim_type == F16 || prim_type == F32) { in EmitErfInv()
971 DCHECK(prim_type == F64); in EmitErfInv()
1045 if (prim_type == F16) { in EmitErfInv()
1051 StatusOr<llvm::Value*> ElementalIrEmitter::EmitErfcInv(PrimitiveType prim_type, in EmitErfcInv() argument
1054 auto type = llvm_ir::PrimitiveTypeToIrType(prim_type, module_); in EmitErfcInv()
1056 return EmitErfInv(prim_type, FSub(one, value)); in EmitErfcInv()
1059 StatusOr<llvm::Value*> ElementalIrEmitter::EmitLog(PrimitiveType prim_type, in EmitLog() argument
[all …]
/external/mesa3d/src/gallium/auxiliary/util/
Du_draw_quad.c47 uint prim_type, in util_draw_vertex_buffer() argument
65 cso_draw_arrays(cso, prim_type, 0, num_verts); in util_draw_vertex_buffer()
68 util_draw_arrays(pipe, prim_type, 0, num_verts); in util_draw_vertex_buffer()
79 uint prim_type, uint num_verts, uint num_attribs) in util_draw_user_vertex_buffer() argument
92 cso_draw_arrays(cso, prim_type, 0, num_verts); in util_draw_user_vertex_buffer()
Du_draw_quad.h47 uint offset, uint prim_type, uint num_attribs,
52 uint prim_type, uint num_verts, uint num_attribs);
/external/mesa3d/src/compiler/glsl/
Dast_type.cpp133 if (qualifier.flags.q.prim_type && new_qualifier.flags.q.prim_type in validate_prim_type()
134 && qualifier.prim_type != new_qualifier.prim_type) { in validate_prim_type()
314 if (q.flags.q.prim_type) { in merge_qualifier()
316 this->flags.q.prim_type = 1; in merge_qualifier()
317 this->prim_type = q.prim_type; in merge_qualifier()
499 if (this->flags.q.prim_type) { in validate_out_qualifier()
501 switch (this->prim_type) { in validate_out_qualifier()
521 valid_out_mask.flags.q.prim_type = 1; in validate_out_qualifier()
593 if (this->flags.q.prim_type) { in validate_in_qualifier()
595 switch (this->prim_type) { in validate_in_qualifier()
[all …]
Dast.h554 unsigned prim_type:1; member
698 GLenum prim_type; member
1258 ast_gs_input_layout(const struct YYLTYPE &locp, GLenum prim_type) in ast_gs_input_layout() argument
1259 : prim_type(prim_type) in ast_gs_input_layout()
1268 const GLenum prim_type;
Dglsl_parser_extras.cpp1753 if (state->in_qualifier->flags.q.prim_type) in set_shader_inout_layout()
1754 shader->info.TessEval.PrimitiveMode = state->in_qualifier->prim_type; in set_shader_inout_layout()
1788 shader->info.Geom.InputType = state->in_qualifier->prim_type; in set_shader_inout_layout()
1793 if (state->out_qualifier->flags.q.prim_type) { in set_shader_inout_layout()
1794 shader->info.Geom.OutputType = state->out_qualifier->prim_type; in set_shader_inout_layout()
Dglsl_parser.yy1287 $$.flags.q.prim_type = 1;
1288 $$.prim_type = map[i].e;
1447 $$.flags.q.prim_type = 1;
1448 $$.prim_type = map[i].e;
Dast_to_hir.cpp3954 if (qual->flags.q.prim_type) { in apply_type_qualifier_to_variable()
4684 num_vertices = vertices_per_prim(state->in_qualifier->prim_type); in handle_geometry_shader_input_decl()
8370 || state->in_qualifier->prim_type == this->prim_type); in hir()
8376 unsigned num_vertices = vertices_per_prim(this->prim_type); in hir()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc236 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestImpl() local
237 Dot(Parameter(&builder, 0, ShapeUtil::MakeShape(prim_type, {2, 2}), "lhs"), in TestImpl()
238 Parameter(&builder, 1, ShapeUtil::MakeShape(prim_type, {2, 2}), "rhs")); in TestImpl()
318 auto prim_type = primitive_util::NativeToPrimitiveType<NativeT>(); in TestImpl() local
322 prim_type, {param.m, param.k}, in TestImpl()
327 prim_type, {param.k, param.n}, in TestImpl()
336 prim_type, {param.m, param.n}, in TestImpl()
485 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestImpl() local
486 Dot(Parameter(&builder, 0, ShapeUtil::MakeShape(prim_type, {2, 3}), "lhs"), in TestImpl()
487 Parameter(&builder, 1, ShapeUtil::MakeShape(prim_type, {3, 2}), "rhs")); in TestImpl()
[all …]
Dmatrix_ops_simple_test.cc195 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestImpl() local
197 ShapeUtil::MakeShape(prim_type, {lhs.height(), lhs.width()}); in TestImpl()
199 ShapeUtil::MakeShape(prim_type, {rhs.height(), rhs.width()}); in TestImpl()
Dfusion_test.cc94 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestElementwise2D() local
102 ShapeUtil::MakeShape(prim_type, {test_width, test_height}); in TestElementwise2D()
132 if (primitive_util::IsFloatingPointType(prim_type)) { in TestElementwise2D()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.cc35 StatusOr<llvm::Value*> CpuElementalIrEmitter::EmitAtan2(PrimitiveType prim_type, in EmitAtan2() argument
40 switch (prim_type) { in EmitAtan2()
72 StatusOr<llvm::Value*> CpuElementalIrEmitter::EmitTanh(PrimitiveType prim_type, in EmitTanh() argument
76 switch (prim_type) { in EmitTanh()
Delemental_ir_emitter.h42 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
44 StatusOr<llvm::Value*> EmitTanh(PrimitiveType prim_type,
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_pt_fetch_shade_pipeline.c363 prim_type(unsigned prim, unsigned flags) in prim_type() function
393 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in fetch_pipeline_run()
421 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in fetch_pipeline_linear_run()
452 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in fetch_pipeline_linear_run_elts()
Ddraw_pt_fetch_shade_pipeline_llvm.c490 prim_type(unsigned prim, unsigned flags) in prim_type() function
520 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in llvm_middle_end_run()
548 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in llvm_middle_end_linear_run()
578 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in llvm_middle_end_linear_run_elts()
/external/mesa3d/src/gallium/drivers/freedreno/a4xx/
Dfd4_draw.h40 static inline uint32_t DRAW4(enum pc_di_primtype prim_type, in DRAW4() argument
44 return CP_DRAW_INDX_OFFSET_0_PRIM_TYPE(prim_type) | in DRAW4()
/external/autotest/client/common_lib/cros/
Ddbus_send.py158 for prim_type, prefix in int_map.iteritems():
159 if isinstance(raw_arg, prim_type):
/external/mesa3d/src/gallium/drivers/nouveau/nv50/
Dnv50_program.h97 uint8_t prim_type; /* point, line strip or tri strip */ member
Dnv50_program.c409 prog->gp.prim_type = NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_LINE_STRIP; in nv50_program_translate()
412 prog->gp.prim_type = NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_TRIANGLE_STRIP; in nv50_program_translate()
417 prog->gp.prim_type = NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_POINTS; in nv50_program_translate()
Dnv50_shader_state.c270 PUSH_DATA (push, gp->gp.prim_type); in nv50_gmtyprog_validate()
276 nv50->state.prim_size = gp->gp.prim_type; /* enum matches vertex count */ in nv50_gmtyprog_validate()
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_util.h101 static inline uint32_t DRAW(enum pc_di_primtype prim_type, in DRAW() argument
106 return (prim_type << 0) | in DRAW()
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_tgsi_vgpu10.c191 VGPU10_PRIMITIVE prim_type;/**< VGPU10 primitive type */ member
1932 emit->gs.prim_type = primType[prop->u[0].Data]; in emit_vgpu10_property()
1933 assert(emit->gs.prim_type != VGPU10_PRIMITIVE_UNDEFINED); in emit_vgpu10_property()
1934 emit->gs.input_size = inputArraySize[emit->gs.prim_type]; in emit_vgpu10_property()
1981 opcode0.primitive = emit->gs.prim_type; in emit_property_instructions()