Searched refs:weight_size (Results 1 – 2 of 2) sorted by relevance
158 const int64 weight_size = example_weights.size(); in Compute() local161 context, weight_size == 1 || weight_size == batch_size, in Compute()179 stream.PushEntry(feature_values(j), (weight_size > 1) in Compute()
446 size_t weight_size = matmul_fwd_pd.get()->weights_desc().get_size(); in CacheWeight() local448 weight_tf_shape.AddDim(weight_size / sizeof(Tweight)); in CacheWeight()455 memcpy(weight_oi_t_data, weight_data, weight_size); in CacheWeight()