Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc221 PrimitiveType prim_type, llvm::Value* value) const { in EmitErfcInv() argument
222 return EmitLibdeviceMathCall("__nv_erfcinv", {value}, {prim_type}, prim_type); in EmitErfcInv()
226 PrimitiveType prim_type, llvm::Value* value) const { in EmitLog() argument
227 return EmitLibdeviceMathCall("__nv_log", {value}, {prim_type}, prim_type); in EmitLog()
231 PrimitiveType prim_type, llvm::Value* value) const { in EmitSin() argument
232 return EmitLibdeviceMathCall("__nv_sin", {value}, {prim_type}, prim_type); in EmitSin()
236 PrimitiveType prim_type, llvm::Value* value) const { in EmitCos() argument
237 return EmitLibdeviceMathCall("__nv_cos", {value}, {prim_type}, prim_type); in EmitCos()
241 PrimitiveType prim_type, llvm::Value* value) const { in EmitExp() argument
242 return EmitLibdeviceMathCall("__nv_exp", {value}, {prim_type}, prim_type); in EmitExp()
[all …]
Delemental_ir_emitter.h61 StatusOr<llvm::Value*> EmitErfcInv(PrimitiveType prim_type,
64 StatusOr<llvm::Value*> EmitLog(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*> EmitPow(PrimitiveType prim_type, llvm::Value* lhs,
79 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc190 auto prim_type = primitive_util::NativeToPrimitiveType<Element>(); in TestSquareMatrixDot() local
192 builder.Parameter(0, ShapeUtil::MakeShape(prim_type, {2, 2}), "lhs"), in TestSquareMatrixDot()
193 builder.Parameter(1, ShapeUtil::MakeShape(prim_type, {2, 2}), "rhs")); in TestSquareMatrixDot()
262 auto prim_type = primitive_util::NativeToPrimitiveType<float>(); in XLA_TEST_P() local
264 builder.Parameter(0, ShapeUtil::MakeShape(prim_type, {param.m, param.k}), in XLA_TEST_P()
266 builder.Parameter(1, ShapeUtil::MakeShape(prim_type, {param.k, param.n}), in XLA_TEST_P()
273 2, ShapeUtil::MakeShape(prim_type, {param.m, param.n}), "addend")); in XLA_TEST_P()
399 auto prim_type = primitive_util::NativeToPrimitiveType<Element>(); in TestNonsquareMatrixDot() local
401 builder.Parameter(0, ShapeUtil::MakeShape(prim_type, {2, 3}), "lhs"), in TestNonsquareMatrixDot()
402 builder.Parameter(1, ShapeUtil::MakeShape(prim_type, {3, 2}), "rhs")); in TestNonsquareMatrixDot()
[all …]
Dmatrix_ops_simple_test.cc175 auto prim_type = primitive_util::NativeToPrimitiveType<float>(); in TEST_P() local
177 ShapeUtil::MakeShape(prim_type, {lhs.height(), lhs.width()}); in TEST_P()
179 ShapeUtil::MakeShape(prim_type, {rhs.height(), rhs.width()}); in TEST_P()
241 auto prim_type = primitive_util::NativeToPrimitiveType<bfloat16>(); in TEST_P() local
243 ShapeUtil::MakeShape(prim_type, {lhs.height(), lhs.width()}); in TEST_P()
245 ShapeUtil::MakeShape(prim_type, {rhs.height(), rhs.width()}); in TEST_P()
Dfusion_test.cc90 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestElementwise2D() local
98 ShapeUtil::MakeShape(prim_type, {test_width, test_height}); in TestElementwise2D()
123 if (primitive_util::IsFloatingPointType(prim_type)) { in TestElementwise2D()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h95 virtual StatusOr<llvm::Value*> EmitErfInv(PrimitiveType prim_type,
98 virtual StatusOr<llvm::Value*> EmitErfcInv(PrimitiveType prim_type,
101 virtual StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type,
105 virtual StatusOr<llvm::Value*> EmitLog(PrimitiveType prim_type,
108 virtual StatusOr<llvm::Value*> EmitSin(PrimitiveType prim_type,
111 virtual StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
114 virtual StatusOr<llvm::Value*> EmitExp(PrimitiveType prim_type,
117 virtual StatusOr<llvm::Value*> EmitPow(PrimitiveType prim_type,
Delemental_ir_emitter.cc870 StatusOr<llvm::Value*> ElementalIrEmitter::EmitErfInv(PrimitiveType prim_type, in EmitErfInv() argument
872 if (prim_type != F32) { in EmitErfInv()
952 PrimitiveType prim_type, llvm::Value* value) const { in EmitErfcInv() argument
954 auto type = llvm_ir::PrimitiveTypeToIrType(prim_type, module_); in EmitErfcInv()
956 return EmitErfInv(prim_type, ir_builder_->CreateFSub(one, value)); in EmitErfcInv()
959 StatusOr<llvm::Value*> ElementalIrEmitter::EmitLog(PrimitiveType prim_type, in EmitLog() argument
965 StatusOr<llvm::Value*> ElementalIrEmitter::EmitSin(PrimitiveType prim_type, in EmitSin() argument
971 StatusOr<llvm::Value*> ElementalIrEmitter::EmitCos(PrimitiveType prim_type, in EmitCos() argument
977 StatusOr<llvm::Value*> ElementalIrEmitter::EmitExp(PrimitiveType prim_type, in EmitExp() argument
983 StatusOr<llvm::Value*> ElementalIrEmitter::EmitPow(PrimitiveType prim_type, in EmitPow() argument
[all …]
/external/mesa3d/src/compiler/glsl/
Dast_type.cpp128 if (qualifier.flags.q.prim_type && new_qualifier.flags.q.prim_type in validate_prim_type()
129 && qualifier.prim_type != new_qualifier.prim_type) { in validate_prim_type()
275 if (q.flags.q.prim_type) { in merge_qualifier()
277 this->flags.q.prim_type = 1; in merge_qualifier()
278 this->prim_type = q.prim_type; in merge_qualifier()
442 if (this->flags.q.prim_type) { in validate_out_qualifier()
444 switch (this->prim_type) { in validate_out_qualifier()
464 valid_out_mask.flags.q.prim_type = 1; in validate_out_qualifier()
536 if (this->flags.q.prim_type) { in validate_in_qualifier()
538 switch (this->prim_type) { in validate_in_qualifier()
[all …]
Dast.h545 unsigned prim_type:1; member
679 GLenum prim_type; member
1225 ast_gs_input_layout(const struct YYLTYPE &locp, GLenum prim_type) in ast_gs_input_layout() argument
1226 : prim_type(prim_type) in ast_gs_input_layout()
1235 const GLenum prim_type;
Dglsl_parser_extras.cpp1732 if (state->in_qualifier->flags.q.prim_type) in set_shader_inout_layout()
1733 shader->info.TessEval.PrimitiveMode = state->in_qualifier->prim_type; in set_shader_inout_layout()
1767 shader->info.Geom.InputType = state->in_qualifier->prim_type; in set_shader_inout_layout()
1772 if (state->out_qualifier->flags.q.prim_type) { in set_shader_inout_layout()
1773 shader->info.Geom.OutputType = state->out_qualifier->prim_type; in set_shader_inout_layout()
Dglsl_parser.yy1290 $$.flags.q.prim_type = 1;
1291 $$.prim_type = map[i].e;
1451 $$.flags.q.prim_type = 1;
1452 $$.prim_type = map[i].e;
/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
91 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/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/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.c478 prim_type(unsigned prim, unsigned flags) in prim_type() function
508 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in llvm_middle_end_run()
536 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in llvm_middle_end_linear_run()
566 prim_info.prim = prim_type(fpme->input_prim, prim_flags); in llvm_middle_end_linear_run_elts()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.cc74 PrimitiveType prim_type, llvm::Value* lhs, llvm::Value* rhs) const { in EmitAtan2() argument
77 switch (prim_type) { in EmitAtan2()
Delemental_ir_emitter.h44 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
/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/ilo/shader/
Dilo_shader_gs.c73 int prim_type; member
164 gcc->vars.prim_type = GEN6_3DPRIM_POINTLIST; in gs_init_vars()
167 gcc->vars.prim_type = GEN6_3DPRIM_LINESTRIP; in gs_init_vars()
170 gcc->vars.prim_type = GEN6_3DPRIM_TRISTRIP; in gs_init_vars()
407 tsrc_imm_d(gcc->vars.prim_type << 2 | in gs_lower_opcode_emit_vue_static()
416 tsrc_imm_d(gcc->vars.prim_type << 2 | in gs_lower_opcode_emit_vue_static()
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_util.h97 static inline uint32_t DRAW(enum pc_di_primtype prim_type, in DRAW() argument
102 return (prim_type << 0) | in DRAW()
/external/mesa3d/src/gallium/drivers/nouveau/nv50/
Dnv50_program.h97 uint8_t prim_type; /* point, line strip or tri strip */ member
Dnv50_program.c404 prog->gp.prim_type = NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_LINE_STRIP; in nv50_program_translate()
407 prog->gp.prim_type = NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_TRIANGLE_STRIP; in nv50_program_translate()
412 prog->gp.prim_type = NV50_3D_GP_OUTPUT_PRIMITIVE_TYPE_POINTS; in nv50_program_translate()
Dnv50_shader_state.c267 PUSH_DATA (push, gp->gp.prim_type); in nv50_gmtyprog_validate()
273 nv50->state.prim_size = gp->gp.prim_type; /* enum matches vertex count */ in nv50_gmtyprog_validate()
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_tgsi_vgpu10.c187 VGPU10_PRIMITIVE prim_type;/**< VGPU10 primitive type */ member
1926 emit->gs.prim_type = primType[prop->u[0].Data]; in emit_vgpu10_property()
1927 assert(emit->gs.prim_type != VGPU10_PRIMITIVE_UNDEFINED); in emit_vgpu10_property()
1928 emit->gs.input_size = inputArraySize[emit->gs.prim_type]; in emit_vgpu10_property()
1975 opcode0.primitive = emit->gs.prim_type; in emit_property_instructions()

12