Searched refs:bias_id (Results 1 – 7 of 7) sorted by relevance
| /external/XNNPACK/src/subgraph/ |
| D | fully-connected.c | 22 uint32_t bias_id, in xnn_define_fully_connected() argument
67 if (bias_id >= subgraph->num_values) { in xnn_define_fully_connected()
70 xnn_node_type_to_string(xnn_node_type_fully_connected), bias_id); in xnn_define_fully_connected()
92 node->inputs[2] = bias_id; in xnn_define_fully_connected()
|
| D | deconvolution-2d.c | 37 uint32_t bias_id, in xnn_define_deconvolution_2d() argument
124 if (bias_id >= subgraph->num_values) { in xnn_define_deconvolution_2d()
127 xnn_node_type_to_string(xnn_node_type_deconvolution_2d), bias_id); in xnn_define_deconvolution_2d()
162 node->inputs[2] = bias_id; in xnn_define_deconvolution_2d()
|
| D | depthwise-convolution-2d.c | 34 uint32_t bias_id, in xnn_define_depthwise_convolution_2d() argument
144 if (bias_id >= subgraph->num_values) { in xnn_define_depthwise_convolution_2d()
147 xnn_node_type_to_string(xnn_node_type_depthwise_convolution_2d), bias_id); in xnn_define_depthwise_convolution_2d()
181 node->inputs[2] = bias_id; in xnn_define_depthwise_convolution_2d()
|
| D | convolution-2d.c | 35 uint32_t bias_id, in xnn_define_convolution_2d() argument
152 if (bias_id >= subgraph->num_values) { in xnn_define_convolution_2d()
155 xnn_node_type_to_string(xnn_node_type_convolution_2d), bias_id); in xnn_define_convolution_2d()
190 node->inputs[2] = bias_id; in xnn_define_convolution_2d()
|
| /external/XNNPACK/test/ |
| D | subgraph-tester.h | 81 uint32_t input_id, uint32_t filter_id, uint32_t bias_id, in add_conv() argument
90 std::numeric_limits<float>::infinity(), input_id, filter_id, bias_id, in add_conv()
104 uint32_t filter_id, uint32_t bias_id, uint32_t output_id) in add_depthwise_conv() argument
112 std::numeric_limits<float>::infinity(), input_id, filter_id, bias_id, in add_depthwise_conv()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 118 int cell_weight_id, int bias_id, in BuildLstmGate() argument
133 RETURN_IF_ERROR(reader->ReadTensor(bias_id, &(fc_attr.bias))); in BuildLstmGate()
226 RETURN_IF_ERROR(reader->ReadTensor(bias_id, &bias)); in BuildLstmGate()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 256 uint32_t bias_id,
311 uint32_t bias_id,
364 uint32_t bias_id,
494 uint32_t filter_id, uint32_t bias_id,
|