Searched defs:num_output_dims (Results 1 – 4 of 4) sorted by relevance
48 const int num_output_dims = output->dims(); in operator() local217 const int num_output_dims = output->dims(); in operator() local
92 const int num_output_dims = output->dims->size; in FillDiagHelper() local
96 const int num_output_dims = output->dims->size; in FillDiagHelper() local
229 int num_output_dims = iter.ndim() - num_reduce_dims; in make_output_calculator() local