Searched refs:new_num_dimensions (Results 1 – 1 of 1) sorted by relevance
2577 const int new_num_dimensions = output_tensor_shape.dim_size(); in SimplifyReduction() local2578 Tensor tensor(DT_INT32, TensorShape({new_num_dimensions})); in SimplifyReduction()2579 for (int i = 0; i < new_num_dimensions; i++) { in SimplifyReduction()