Searched refs:weights_size (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | fuse_binary_into_following_affine.cc | 82 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine() local
83 const int weights_per_depth = weights_size / output_depth; in FuseAddOrSubParamsIntoFollowingAffine()
84 CHECK_EQ(weights_size, weights_per_depth * output_depth); in FuseAddOrSubParamsIntoFollowingAffine()
97 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine() local
98 const int weights_per_depth = weights_size / output_depth; in FuseAddOrSubParamsIntoFollowingAffine()
99 CHECK_EQ(weights_size, weights_per_depth * output_depth); in FuseAddOrSubParamsIntoFollowingAffine()
139 const int weights_size = RequiredBufferSizeForShape(weights.shape()); in FuseMulOrDivParamsIntoFollowingAffine() local
140 for (int i = 0; i < weights_size; i++) { in FuseMulOrDivParamsIntoFollowingAffine()
|
| D | fuse_binary_into_preceding_affine.cc | 162 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseMulOrDivParamsIntoPrecedingAffine() local
163 const int weights_per_depth = weights_size / output_depth; in FuseMulOrDivParamsIntoPrecedingAffine()
164 CHECK_EQ(weights_size, weights_per_depth * output_depth); in FuseMulOrDivParamsIntoPrecedingAffine()
|
| /external/XNNPACK/test/ |
| D | weights-cache.cc | 236 const size_t weights_size = weights.size(); in TEST() local
239 xnn_get_or_insert_weights_cache(&cache, cache_end(&cache), weights_size); in TEST()
254 ASSERT_EQ(weights_size, cache.cache.weights.size); in TEST()
266 const size_t weights_size = weights.size(); in TEST() local
272 xnn_get_or_insert_weights_cache(&cache, cache_end(&cache), weights_size); in TEST()
275 ASSERT_LE(num_threads, weights_size); in TEST()
288 ASSERT_EQ(weights_size * num_threads, cache.cache.weights.size); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op_gpu.cu.cc | 145 const int weights_size, T* out, in BincountColReduceKernel() argument
157 T value = (weights_size == 0) ? T(1) : ldg(weights + index); in BincountColReduceKernel()
166 const int weights_size, T* out, in BincountColReduceSharedKernel() argument
187 T value = (weights_size == 0) ? T(1) : ldg(weights + index); in BincountColReduceSharedKernel()
|
| D | bincount_op.cc | 386 const int64_t weights_size = weights.size(); in Compute() local
441 if (weights_size) { in Compute()
482 const int64_t weights_size = weights.size(); in Compute() local
526 T value = (weights_size > 0) ? weights(idx) : T(1); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bincount_op.cc | 94 auto weights_size = weights_shape.dimensions(0); in Compile() local
96 if (weights_size) { in Compile()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-002.md | 32 T value = (weights_size > 0) ? weights(idx) : T(1);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.cc | 283 int weights_size = conv_weights_count * SizeOf(data_type); in IsNodeSupported() local
284 if (convs_count_ >= 3 || buffer_size_ + weights_size > 1024 * 3) { in IsNodeSupported()
|