Searched refs:src_weights (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | winograd_util.cc | 112 const Tensor<OHWI, DataType::FLOAT32>& src_weights, in RearrangeWeightsToWinograd4x4To6x6Weights() argument
115 dst_shape.o = src_weights.shape.o; in RearrangeWeightsToWinograd4x4To6x6Weights()
118 dst_shape.i = src_weights.shape.i; in RearrangeWeightsToWinograd4x4To6x6Weights()
130 for (int d = 0; d < src_weights.shape.o; ++d) { in RearrangeWeightsToWinograd4x4To6x6Weights()
131 for (int s = 0; s < src_weights.shape.i; ++s) { in RearrangeWeightsToWinograd4x4To6x6Weights()
135 const int f_index = src_weights.shape.LinearIndex({d, y, x, s}); in RearrangeWeightsToWinograd4x4To6x6Weights()
136 in_vals[y * 3 + x] = src_weights.data[f_index]; in RearrangeWeightsToWinograd4x4To6x6Weights()
|
| D | winograd_util.h | 39 const Tensor<OHWI, DataType::FLOAT32>& src_weights,
|
| /external/libyuv/files/source/ |
| D | row_neon64.cc | 2973 void FloatDivToByteRow_NEON(const float* src_weights, in FloatDivToByteRow_NEON() argument
3006 : "+r"(src_weights), // %0 in FloatDivToByteRow_NEON()
|
| D | row_common.cc | 3335 void FloatDivToByteRow_C(const float* src_weights, in FloatDivToByteRow_C() argument
3342 dst_out[x] = Clamp(src_values[x] / src_weights[x]); in FloatDivToByteRow_C()
3343 dst_mask[x] = src_weights[x] > 0 ? 0 : 0xff; in FloatDivToByteRow_C()
|
| /external/libyuv/files/unit_test/ |
| D | planar_test.cc | 3280 align_buffer_page_end(src_weights, kPixels * 4); in TestFloatDivToByte()
3291 (reinterpret_cast<float*>(src_weights))[i] = scale; in TestFloatDivToByte()
3300 FloatDivToByteRow_C(reinterpret_cast<float*>(src_weights), in TestFloatDivToByte()
3307 FloatDivToByteRow_NEON(reinterpret_cast<float*>(src_weights), in TestFloatDivToByte()
3311 FloatDivToByteRow_C(reinterpret_cast<float*>(src_weights), in TestFloatDivToByte()
3316 FloatDivToByteRow_C(reinterpret_cast<float*>(src_weights), in TestFloatDivToByte()
3331 free_aligned_buffer_page_end(src_weights); in TestFloatDivToByte()
|
| /external/libyuv/files/include/libyuv/ |
| D | row.h | 4021 void FloatDivToByteRow_C(const float* src_weights,
4026 void FloatDivToByteRow_NEON(const float* src_weights,
|