Searched refs:range_output_func (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | split_op.cc | 177 auto range_output_func = [&indices, context, &output_shape, prefix_dim_size,
213 num_split, input_element_count / num_split, range_output_func);
216 range_output_func(0, num_split);
|
| H A D | split_v_op.cc | 235 auto range_output_func = [&indices, context, &input_shape, prefix_dim_size, variable
271 num_split, input_element_count / num_split, range_output_func);
274 range_output_func(0, num_split);
|
Completed in 129 milliseconds