Searched refs:outputs_count (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/ |
| D | hexagon_controller.c | 464 int outputs_count) { in hexagon_controller_AppendNode() argument
483 for (int i = 0; i < outputs_count; ++i) { in hexagon_controller_AppendNode()
489 padding_id, inputs_count, outputs_count, input_param_buf, in hexagon_controller_AppendNode()
493 inputs_count, outputs_count, input_param_buf, output_param_buf); in hexagon_controller_AppendNode()
497 inputs_count, outputs, outputs_count); in hexagon_controller_AppendNode()
|
| /external/mesa3d/src/gallium/drivers/freedreno/ir3/ |
| D | ir3_shader.h | 259 unsigned outputs_count; member
415 for (j = 0; j < so->outputs_count; j++) in ir3_find_output()
438 for (j = 0; j < so->outputs_count; j++) in ir3_find_output()
506 for (j = 0; j < so->outputs_count; j++) in ir3_find_output_regid()
|
| D | ir3_shader.c | 88 for (i = 0; i < v->outputs_count; i++) { in fixup_regfootprint()
434 for (i = 0; i < so->outputs_count; i++) { in ir3_shader_disasm()
454 for (i = 0; i < so->outputs_count; i++) { in ir3_shader_disasm()
|
| D | ir3_compiler_nir.c | 3056 so->outputs_count = MAX2(so->outputs_count, n + 1); in setup_output()
3287 for (i = 0, j = 0; i < so->outputs_count; i++) { in ir3_compile_shader_nir()
3302 so->outputs_count = j; in ir3_compile_shader_nir()
3385 for (i = 0; i < so->outputs_count; i++) { in ir3_compile_shader_nir()
|
| /external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_soc_interface/ |
| D | soc_interface.c | 229 int outputs_count) { in soc_interface_AppendNode() argument
233 inputs_count, (hexagon_nn_output*)outputs, outputs_count); in soc_interface_AppendNode()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | soc_interface.h | 96 int outputs_count);
|
| D | hexagon_control_wrapper.cc | 192 int outputs_count = 0; in SetupGraph() local
200 outputs_count += output_params.max_byte_size_size(); in SetupGraph()
204 outputs_count); in SetupGraph()
|
| /external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_soc_interface/include/ |
| D | soc_interface.h | 97 int outputs_count);
|
| /external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/include/ |
| D | hexagon_controller.h | 133 int outputs_count);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 665 int axis, int outputs_count, Scalar* const* output_data, in TensorFlowSplit() argument
667 std::vector<RuntimeShape> output_shapes(outputs_count); in TensorFlowSplit()
668 std::vector<const RuntimeShape*> output_shapes_indirect(outputs_count); in TensorFlowSplit()
669 for (int i = 0; i < outputs_count; ++i) { in TensorFlowSplit()
675 op_params.num_split = outputs_count; in TensorFlowSplit()
683 int outputs_count, Scalar* const* output_data, in TensorFlowSplit() argument
685 TFLITE_DCHECK_GE(outputs_count, 1); in TensorFlowSplit()
686 for (int i = 0; i < outputs_count; i++) { in TensorFlowSplit()
694 TensorFlowSplit(input_data, input_dims, /*axis=*/0, outputs_count, in TensorFlowSplit()
1067 int dimensions, int outputs_count, Scalar* const* output_datas, in Unpack() argument
[all …]
|
| D | reference_ops.h | 1746 const int outputs_count = params.num_split; in Unpack() local
1764 for (int i = 0; i < outputs_count; ++i) { in Unpack()
1767 int loc = k * outputs_count * copy_size + i * copy_size; in Unpack()
2233 int outputs_count = params.num_split; in Split() local
2237 for (int i = 0; i < outputs_count; i++) { in Split()
2260 for (int i = 0; i < outputs_count; ++i) { in Split()
|