Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconcatenation.h34 int inputs_count = params.inputs_count; in Concatenation() local
39 for (int i = 0; i < inputs_count; i++) { in Concatenation()
62 for (int i = 0; i < inputs_count; ++i) { in Concatenation()
82 int inputs_count = params.inputs_count; in ConcatenationWithScaling() local
90 for (int i = 0; i < inputs_count; i++) { in ConcatenationWithScaling()
114 for (int i = 0; i < inputs_count; ++i) { in ConcatenationWithScaling()
Dlegacy_reference_ops.h649 const Dims<4>* const* input_dims, int inputs_count, in Concatenation() argument
654 std::vector<RuntimeShape> input_shapes(inputs_count); in Concatenation()
655 std::vector<const RuntimeShape*> input_shapes_indirect(inputs_count); in Concatenation()
656 for (int i = 0; i < inputs_count; ++i) { in Concatenation()
662 op_params.inputs_count = inputs_count; in Concatenation()
671 const float* input_scale, int inputs_count, in Concatenation() argument
675 std::vector<RuntimeShape> input_shapes(inputs_count); in Concatenation()
676 std::vector<const RuntimeShape*> input_shapes_indirect(inputs_count); in Concatenation()
677 for (int i = 0; i < inputs_count; ++i) { in Concatenation()
685 op_params.inputs_count = inputs_count; in Concatenation()
[all …]
Dreference_ops.h323 int inputs_count = params.inputs_count; in Pack() local
335 for (int i = 0; i < inputs_count; ++i) { in Pack()
338 int loc = k * inputs_count * copy_size + i * copy_size; in Pack()
387 int inputs_count = params.inputs_count; in PackWithScaling() local
404 for (int i = 0; i < inputs_count; ++i) { in PackWithScaling()
511 concat_params.inputs_count = concat_input_arrays_data.size(); in LstmCell()
723 concat_params.inputs_count = 2; in LstmCell()
/external/mesa3d/src/gallium/drivers/freedreno/a2xx/
Dir2_nir.c516 instr->src[0] = ir2_src(ctx->f->inputs_count, 0, IR2_SRC_INPUT); in load_input()
571 for (idx = 0; idx < ctx->f->inputs_count; idx++) in store_output()
574 if (idx == ctx->f->inputs_count) in store_output()
631 tmp->src[0] = ir2_src(ctx->f->inputs_count, 0, IR2_SRC_INPUT); in emit_intrinsic()
642 instr->src[0] = ir2_src(ctx->f->inputs_count, IR2_SWIZZLE_ZW, IR2_SRC_INPUT); in emit_intrinsic()
765 n = ctx->f->inputs_count++; in setup_input()
1129 ctx->f->inputs_count = 0; in ir2_nir_compile()
1139 for (idx = 0; idx < ctx->f->inputs_count; idx++) { in ir2_nir_compile()
1172 ctx->input[ctx->f->inputs_count].initialized = false; in ir2_nir_compile()
Dir2.h68 unsigned inputs_count; member
Dfd2_program.c252 vs_export = MAX2(1, f->inputs_count) - 1; in fd2_program_emit()
264 COND(fp, A2XX_SQ_CONTEXT_MISC_PARAM_GEN_POS(f->inputs_count)) | in fd2_program_emit()
Dir2_assemble.c379 if (ctx->so->type == MESA_SHADER_VERTEX && ctx->f->inputs_count == 0) { in assemble()
485 alloc.size = ctx->f->inputs_count - 1; in assemble()
/external/mesa3d/src/freedreno/ir3/
Dir3_shader.h589 unsigned inputs_count; member
840 while (++i < so->inputs_count) in ir3_next_varying()
922 if (j >= fs->inputs_count) in ir3_link_shaders()
973 for (j = 0; j < so->inputs_count; j++) in ir3_find_sysval_regid()
Dir3_compiler_nir.c1338 unsigned n = so->inputs_count++; in add_sysval_input_compmask()
2946 so->inputs_count = MAX2(so->inputs_count, n + 1); in setup_input()
3008 uint8_t used_components[so->inputs_count]; in pack_inlocs()
3024 compile_assert(ctx, i < so->inputs_count); in pack_inlocs()
3033 compile_assert(ctx, i < so->inputs_count); in pack_inlocs()
3059 for (unsigned i = 0; i < so->inputs_count; i++) { in pack_inlocs()
3268 ctx->so->inputs_count = ctx->s->num_inputs; in emit_instructions()
3612 debug_assert(n < so->nonbinning->inputs_count); in ir3_compile_shader_nir()
3760 for (unsigned i = 0; i < so->inputs_count; i++) in ir3_compile_shader_nir()
Dir3_shader.c62 for (i = 0; i < v->inputs_count; i++) { in fixup_regfootprint()
599 for (i = 0; i < so->inputs_count; i++) { in ir3_shader_disasm()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dsoc_interface.h95 int inputs_count, const void* const outputs,
Dhexagon_control_wrapper.cc191 int inputs_count = 0; in SetupGraph() local
195 inputs_count += input_params.node_input_size(); in SetupGraph()
203 soc_interface_AllocateNodeInputAndNodeOutputArray(inputs_count, in SetupGraph()
/external/tensorflow/tensorflow/lite/kernels/
Dconcatenation.cc131 op_params.inputs_count = node->inputs->size; \ in Eval()
150 op_params.inputs_count = node->inputs->size; \ in Eval()
Dpack.cc103 op_params.inputs_count = values_count; in PackImpl()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dconcatenation.cc188 data->params.inputs_count = node->inputs->size; in Prepare()
194 data->params.inputs_count = node->inputs->size; in Prepare()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dgru_cell.cc58 concat_params.inputs_count = concat_arrays_data.size(); in GruCell()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc463 const size_t inputs_count = input_layers.size(); in Main() local
532 for (int n = 0; n < inputs_count; ++n) { in Main()
/external/mesa3d/src/gallium/drivers/freedreno/a6xx/
Dfd6_program.c902 for (uint32_t i = 0; i < vs->inputs_count; i++) in setup_stateobj()
983 for (int j = -1; (j = ir3_next_varying(fs, j)) < (int)fs->inputs_count; ) { in emit_interp_state()
1074 for (unsigned i = 0; i < bs->inputs_count; i++) { in fd6_program_create()
/external/mesa3d/src/gallium/drivers/freedreno/a3xx/
Dfd3_program.c355 for (j = -1; (j = ir3_next_varying(fp, j)) < (int)fp->inputs_count; ) { in fd3_program_emit()
Dfd3_emit.c389 for (i = 0; i < vp->inputs_count; i++) { in fd3_emit_vertex_bufs()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h946 uint16 inputs_count;
1113 uint16 inputs_count;
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h834 uint16_t inputs_count; member
1014 uint16_t inputs_count; member
/external/mesa3d/src/gallium/drivers/freedreno/a4xx/
Dfd4_program.c459 for (j = -1; (j = ir3_next_varying(s[FS].v, j)) < (int)s[FS].v->inputs_count; ) { in fd4_program_emit()
/external/mesa3d/src/gallium/drivers/freedreno/a5xx/
Dfd5_program.c605 for (j = -1; (j = ir3_next_varying(s[FS].v, j)) < (int)s[FS].v->inputs_count; ) { in fd5_program_emit()
/external/mesa3d/src/freedreno/vulkan/
Dtu_pipeline.c1160 (i = ir3_next_varying(fs, i)) < (int) fs->inputs_count;) { in tu6_emit_vpc_varying_modes()
1583 for (input_idx = 0; input_idx < vs->inputs_count; input_idx++) { in tu6_emit_vertex_input()
1589 if (input_idx == vs->inputs_count) in tu6_emit_vertex_input()

12