HomeSort by relevance Sort by last modified time
    Searched refs:neg_one_dim (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
split_op.cc 130 int neg_one_dim = -1; variable
149 ctx, neg_one_dim == -1,
153 neg_one_dim = i;
161 (neg_one_dim == -1 &&
163 (neg_one_dim >= 0 &&
172 if (neg_one_dim >= 0) {
173 split_sizes[neg_one_dim] =
  /external/tensorflow/tensorflow/core/kernels/
split_v_op.cc 111 int neg_one_dim = -1; local
117 OP_REQUIRES(context, neg_one_dim == -1,
120 neg_one_dim = d;
128 (neg_one_dim == -1 && determined_size == input_size_split_dim) ||
129 (neg_one_dim >= 0 && determined_size <= input_size_split_dim),
137 if (neg_one_dim >= 0) {
138 (*split_sizes_vec)[neg_one_dim] = input_size_split_dim - determined_size;

Completed in 869 milliseconds