Home
last modified time | relevance | path

Searched defs:weights_size (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_following_affine.cc82 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine() local
97 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseAddOrSubParamsIntoFollowingAffine() local
139 const int weights_size = RequiredBufferSizeForShape(weights.shape()); in FuseMulOrDivParamsIntoFollowingAffine() local
Dfuse_binary_into_preceding_affine.cc162 const int weights_size = RequiredBufferSizeForShape(weights_shape); in FuseMulOrDivParamsIntoPrecedingAffine() local
/external/tensorflow/tensorflow/core/kernels/
Dbincount_op_gpu.cu.cc134 const int weights_size, T* out, in BincountColReduceKernel()
155 const int weights_size, T* out, in BincountColReduceSharedKernel()
Dbincount_op.cc365 const int64_t weights_size = weights.size(); in Compute() local
445 const int64_t weights_size = weights.size(); in Compute() local