Searched defs:weights_size (Results 1 – 5 of 5) 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
134 const int weights_size, T* out, in BincountColReduceKernel()155 const int weights_size, T* out, in BincountColReduceSharedKernel()
332 const int64 weights_size = weights.size(); in Compute() local412 const int64 weights_size = weights.size(); in Compute() local
120 int weights_size = mli_hlp_count_elem_num(weights, 0) * in get_arc_scratch_buffer_for_conv_tensors() local185 int weights_size = mli_hlp_count_elem_num(weights, 0) * in get_arc_scratch_buffer_for_fully_connect_tensors() local