Searched defs:featureSize (Results 1 – 8 of 8) sorted by relevance
57 private final int featureSize; field in Network94 int featureSize, in Network()133 int featureSize) { in Network()464 private final int featureSize; field in Network.SerializationProxy478 int featureSize, in SerializationProxy()
163 final int featureSize = target.size(); in predict() local177 final int featureSize = target.size(); in update() local
65 int64_t featureSize, int64_t weight_stride0, int64_t weight_stride1, in EmbeddingBag_updateOutputKernel_max()115 int64_t featureSize, int64_t weight_stride0, int64_t weight_stride1, in EmbeddingBag_updateOutputKernel_sum_mean()374 int64_t featureSize = weight.size(1); in _embedding_bag_cuda() local
META-INF/ META-INF/MANIFEST.MF android/ android/ext/ android/ ...
1074 int64_t featureSize = weight.size(1); in embedding_bag_cpu_max_out() local
META-INF/ META-INF/MANIFEST.MF META-INF/NOTICE.txt META ...
private_join_and_compute/util/proto/test/Test.class Test.java package private_join_and_compute.util. ...