Home
last modified time | relevance | path

Searched refs:weights_cache (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dweights_cache_test.cc44 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreateWithSize(four_mb), in TEST() local
49 delegate_options.weights_cache = weights_cache.get(); in TEST()
58 TfLiteXNNPackDelegateWeightsCacheFinalizeHard(weights_cache.get())); in TEST()
74 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreate(), in TEST() local
79 delegate_options.weights_cache = weights_cache.get(); in TEST()
102 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreate(), in TEST() local
107 delegate_options.weights_cache = weights_cache.get(); in TEST()
114 TfLiteXNNPackDelegateWeightsCacheFinalizeHard(weights_cache.get())); in TEST()
136 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreate(), in TEST() local
141 delegate_options.weights_cache = weights_cache.get(); in TEST()
[all …]
Dprelu_tester.h90 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
91 weights_cache_ = weights_cache; in WeightsCache()
Dfully_connected_tester.h124 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
125 weights_cache_ = weights_cache; in WeightsCache()
Dquantized_fully_connected_tester.h151 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
152 weights_cache_ = weights_cache; in WeightsCache()
Dtranspose_conv_tester.h178 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
179 weights_cache_ = weights_cache; in WeightsCache()
Dquantized_transpose_conv_tester.h170 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
171 weights_cache_ = weights_cache; in WeightsCache()
Ddepthwise_conv_2d_tester.h222 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
223 weights_cache_ = weights_cache; in WeightsCache()
Dconv_2d_tester.h231 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
232 weights_cache_ = weights_cache; in WeightsCache()
Dxnnpack_delegate.h46 struct TfLiteXNNPackDelegateWeightsCache* weights_cache; member
Dquantized_conv_2d_tester.h256 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
257 weights_cache_ = weights_cache; in WeightsCache()
Dquantized_depthwise_conv_2d_tester.h255 TfLiteXNNPackDelegateWeightsCache* weights_cache) { in WeightsCache() argument
256 weights_cache_ = weights_cache; in WeightsCache()
Dfully_connected_test.cc494 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreate(), in TEST() local
496 delegate_options.weights_cache = weights_cache.get(); in TEST()
515 .WeightsCache(weights_cache.get()) in TEST()
Dsigned_quantized_fully_connected_test.cc437 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreate(), in TEST() local
439 delegate_options.weights_cache = weights_cache.get(); in TEST()
464 .WeightsCache(weights_cache.get()) in TEST()
Dprelu_test.cc629 weights_cache(TfLiteXNNPackDelegateWeightsCacheCreate(), in TEST() local
631 delegate_options.weights_cache = weights_cache.get(); in TEST()
648 .WeightsCache(weights_cache.get()) in TEST()
/external/XNNPACK/test/
Dprelu-operator-tester.h149 .weights_cache = NULL, in TestF16()
151 xnn_weights_cache weights_cache; in TestF16() local
153 xnn_init_weights_cache(&weights_cache); in TestF16()
154 caches.weights_cache = &weights_cache; in TestF16()
173 … xnn_finalize_weights_cache(&weights_cache, xnn_weights_cache_finalization_kind_soft)); in TestF16()
193 const size_t old_weights_cache_size = weights_cache.cache.weights.size; in TestF16()
217 VerifyWeightsCache(weights_cache, old_weights_cache_size); in TestF16()
218 xnn_release_weights_cache(&weights_cache); in TestF16()
265 .weights_cache = NULL, in TestF32()
267 xnn_weights_cache weights_cache; in TestF32() local
[all …]
Dfully-connected-operator-tester.h238 .weights_cache = NULL, in TestQS8()
240 xnn_weights_cache weights_cache; in TestQS8() local
242 xnn_init_weights_cache(&weights_cache); in TestQS8()
243 caches.weights_cache = &weights_cache; in TestQS8()
263 … xnn_finalize_weights_cache(&weights_cache, xnn_weights_cache_finalization_kind_soft)); in TestQS8()
285 size_t old_weights_cache_size = weights_cache.cache.weights.size; in TestQS8()
315 VerifyWeightsCache(weights_cache, old_weights_cache_size); in TestQS8()
316 xnn_release_weights_cache(&weights_cache); in TestQS8()
419 .weights_cache = NULL, in TestQU8()
421 xnn_weights_cache weights_cache; in TestQU8() local
[all …]
Ddeconvolution-operator-tester.h541 .weights_cache = NULL, in TestQS8()
543 xnn_weights_cache weights_cache; in TestQS8() local
545 xnn_init_weights_cache(&weights_cache); in TestQS8()
546 caches.weights_cache = &weights_cache; in TestQS8()
564 … xnn_finalize_weights_cache(&weights_cache, xnn_weights_cache_finalization_kind_soft)); in TestQS8()
584 size_t old_weights_cache_size = weights_cache.cache.weights.size; in TestQS8()
614 VerifyWeightsCache(&weights_cache, old_weights_cache_size); in TestQS8()
616 xnn_release_weights_cache(&weights_cache); in TestQS8()
646 void VerifyWeightsCache(xnn_weights_cache* weights_cache, size_t old_size) const { in VerifyWeightsCache() argument
647 ASSERT_EQ(weights_cache->cache.hits, 1); in VerifyWeightsCache()
[all …]
Dconvolution-operator-tester.h671 .weights_cache = NULL, in TestNHWCxQC8()
673 xnn_weights_cache weights_cache; in TestNHWCxQC8() local
675 xnn_init_weights_cache(&weights_cache); in TestNHWCxQC8()
676 caches.weights_cache = &weights_cache; in TestNHWCxQC8()
700 … xnn_finalize_weights_cache(&weights_cache, xnn_weights_cache_finalization_kind_soft)); in TestNHWCxQC8()
721 size_t old_weights_cache_size = weights_cache.cache.weights.size; in TestNHWCxQC8()
754 VerifyWeightsCache(weights_cache, old_weights_cache_size); in TestNHWCxQC8()
755 xnn_release_weights_cache(&weights_cache); in TestNHWCxQC8()
887 .weights_cache = NULL, in TestNHWCxQS8()
889 xnn_weights_cache weights_cache; in TestNHWCxQS8() local
[all …]
/external/XNNPACK/src/
Druntime.c76 struct xnn_weights_cache* weights_cache = NULL; in xnn_create_weights_cache_with_size() local
84 weights_cache = xnn_allocate_zero_memory(sizeof(struct xnn_weights_cache)); in xnn_create_weights_cache_with_size()
85 if (weights_cache == NULL) { in xnn_create_weights_cache_with_size()
90 status = xnn_init_weights_cache_with_size(weights_cache, size); in xnn_create_weights_cache_with_size()
94 *weights_cache_out = weights_cache; in xnn_create_weights_cache_with_size()
98 xnn_release_weights_cache(weights_cache); in xnn_create_weights_cache_with_size()
107 enum xnn_status xnn_delete_weights_cache(xnn_weights_cache_t weights_cache) in xnn_delete_weights_cache() argument
109 enum xnn_status status = xnn_release_weights_cache(weights_cache); in xnn_delete_weights_cache()
113 xnn_release_memory(weights_cache); in xnn_delete_weights_cache()
135 xnn_weights_cache_t weights_cache, in xnn_create_runtime_v3() argument
[all …]
Doperator-delete.c30 if (op->weights_cache == NULL) { in xnn_delete_operator()
/external/XNNPACK/src/operators/
Dprelu-nc.c89 prelu_op->weights_cache = caches->weights_cache; in create_prelu_nc()
99 prelu_op->weights_cache, weights_ptr, aligned_total_weights_size); in create_prelu_nc()
199 if (prelu_op->weights_cache != NULL && !xnn_weights_cache_is_finalized(prelu_op->weights_cache)) { in setup_prelu_nc()
Dfully-connected-nc.c111 fully_connected_op->weights_cache = caches->weights_cache; in create_fully_connected_nc()
151 fully_connected_op->weights_cache, weights_ptr, aligned_total_weights_size); in create_fully_connected_nc()
222 if (fully_connected_op->weights_cache != NULL && in setup_fully_connected_nc()
223 !xnn_weights_cache_is_finalized(fully_connected_op->weights_cache)) { in setup_fully_connected_nc()
Dconvolution-nchw.c223 convolution_op->weights_cache = caches->weights_cache; in xnn_create_convolution2d_nchw_f32()
432 convolution_op->weights_cache, weights_ptr, aligned_total_weights_size); in xnn_create_convolution2d_nchw_f32()
472 convolution_op->weights_cache, weights_ptr, aligned_total_weights_size); in xnn_create_convolution2d_nchw_f32()
558 …if (convolution_op->weights_cache != NULL && !xnn_weights_cache_is_finalized(convolution_op->weigh… in setup_convolution2d_nchw()
/external/XNNPACK/src/xnnpack/
Doperator.h303 struct xnn_weights_cache* weights_cache; member
308 if (op->weights_cache == NULL) { in packed_weights()
311 return (void*) ((uintptr_t) op->weights_cache->cache.weights.start + op->packed_weights.offset); in packed_weights()
316 return op->weights_cache != NULL; in use_weights_cache()
Dcache.h111 struct xnn_weights_cache *weights_cache; member

12