Searched refs:fixed_dim (Results 1 – 2 of 2) sorted by relevance
652 for (int fixed_dim=0;fixed_dim<2;++fixed_dim) { in get_max_sizes() local655 double M = maximum_sizes[fixed_dim]; in get_max_sizes()658 int x0_dim = !fixed_dim; in get_max_sizes()662 sizes[(*numberOfSizes)][fixed_dim] = (size_t)M; in get_max_sizes()672 for (int fixed_dim=0;fixed_dim<3;++fixed_dim) { in get_max_sizes() local675 double M = maximum_sizes[fixed_dim]; in get_max_sizes()679 int x0_dim = (fixed_dim == 0) ? 1 : 0; in get_max_sizes()680 int x1_dim = (fixed_dim == 2) ? 1 : 2; in get_max_sizes()702 sizes[(*numberOfSizes)][fixed_dim] = (size_t)M; in get_max_sizes()
3978 def _expand_mask(mask_t, input_t, fixed_dim=1): argument3986 multiples = [1] * fixed_dim + input_t.shape.as_list()[fixed_dim:]4054 _expand_mask(mask, outputs, fixed_dim=2), outputs,4132 _expand_mask(mask_t, o, fixed_dim=len(mask_t.shape))