Searched defs:weights_size (Results 1 – 8 of 8) sorted by relevance
82 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine() local97 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine() local139 const int weights_size = RequiredBufferSizeForShape(weights.shape()); in FuseMulOrDivParamsIntoFollowingAffine() local
162 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseMulOrDivParamsIntoPrecedingAffine() local
145 const int weights_size, T* out, in BincountColReduceKernel()166 const int weights_size, T* out, in BincountColReduceSharedKernel()
386 const int64_t weights_size = weights.size(); in Compute() local482 const int64_t weights_size = weights.size(); in Compute() local
236 const size_t weights_size = weights.size(); in TEST() local266 const size_t weights_size = weights.size(); in TEST() local
94 auto weights_size = weights_shape.dimensions(0); in Compile() local
283 int weights_size = conv_weights_count * SizeOf(data_type); in IsNodeSupported() local
1230 unsigned weights_size; in calculate_weight_bo_size() local