Searched refs:tensor_sizes (Results 1 – 2 of 2) sorted by relevance
239 tensor_sizes = [17, 32, 81, 12, 15, 23, 22] * 5240 tensor_sizes = tensor_sizes[:size]242 tensor = tf.ones([tensor_sizes[rank]] + [17] * (dim - 1),247 expected_size = sum(tensor_sizes)252 rank_size = [tensor_sizes[i]] + [17] * (dim - 1)254 [sum(tensor_sizes[:i])] + [0] * (dim - 1),
52 std::vector<int64> tensor_sizes(num_spatial_dims + 2, 1); in ExpandWithBatchAndFeatureDimensions() local55 tensor_sizes[dim] = spatial_dim_sizes[i]; in ExpandWithBatchAndFeatureDimensions()57 return tensor_sizes; in ExpandWithBatchAndFeatureDimensions()