Home
last modified time | relevance | path

Searched refs:b0 (Results 1 – 2 of 2) sorted by relevance

/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/
Dbatch_matmul.h81 for (int b0 = 0; b0 < batch_dim0; ++b0) { in BatchMatMul() local
82 const Ta* lhs_ptr0 = lhs_data + (b0 * lhs_ext0); in BatchMatMul()
83 const Tb* rhs_ptr0 = rhs_data + (b0 * rhs_ext0); in BatchMatMul()
90 Tout* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul()
158 for (int b0 = 0; b0 < batch_dim0; ++b0) { in BatchMatMul() local
159 const int8_t* lhs_ptr0 = lhs_data + (b0 * lhs_ext0); in BatchMatMul()
160 const int8_t* rhs_ptr0 = rhs_data + (b0 * rhs_ext0); in BatchMatMul()
161 const int32_t* ioff_ptr0 = input_offset + (b0 * ioff_ext0); in BatchMatMul()
162 const float* scale_ptr0 = scaling_factors + (b0 * ioff_ext0); in BatchMatMul()
163 const int32_t* woff_ptr0 = row_sums + (b0 * woff_ext0); in BatchMatMul()
[all …]
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/
Dcommon.h92 auto b0 = vld1q_f32(bias_data + i); in BiasAndClamp() local
100 auto x0 = vaddq_f32(a0, b0); in BiasAndClamp()