Searched refs:reduction_indices_shape (Results 1 – 2 of 2) sorted by relevance
68 const TensorShapeProto& reduction_indices_shape = in Optimize() local70 if (NumCoefficients(reduction_indices_shape) == 1) { in Optimize()
701 PartialTensorShape reduction_indices_shape(reduction_indices_prop.shape()); in MaterializeReductionIndices() local702 const int num_reduction_indices = reduction_indices_shape.num_elements(); in MaterializeReductionIndices()