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

  /external/tensorflow/tensorflow/core/kernels/
split_op.cc 159 auto range_output_func = [&indices, context, &output_shape, prefix_dim_size,
193 num_split, input_element_count / num_split, range_output_func);
196 range_output_func(0, num_split);
split_v_op.cc 212 auto range_output_func = [&indices, context, &input_shape, split_dim, local
247 num_split, input_element_count / num_split, range_output_func);
250 range_output_func(0, num_split);

Completed in 178 milliseconds