Searched refs:neg_one_dim (Results 1 – 2 of 2) sorted by relevance
130 int neg_one_dim = -1; in Compile() local149 ctx, neg_one_dim == -1, in Compile()153 neg_one_dim = i; in Compile()161 (neg_one_dim == -1 && in Compile()163 (neg_one_dim >= 0 && in Compile()172 if (neg_one_dim >= 0) { in Compile()173 split_sizes[neg_one_dim] = in Compile()
111 int neg_one_dim = -1; in ComputeEasyCases() local117 OP_REQUIRES(context, neg_one_dim == -1, in ComputeEasyCases()120 neg_one_dim = d; in ComputeEasyCases()128 (neg_one_dim == -1 && determined_size == input_size_split_dim) || in ComputeEasyCases()129 (neg_one_dim >= 0 && determined_size <= input_size_split_dim), in ComputeEasyCases()137 if (neg_one_dim >= 0) { in ComputeEasyCases()138 (*split_sizes_vec)[neg_one_dim] = input_size_split_dim - determined_size; in ComputeEasyCases()