Searched refs:reduction_indices_shape (Results 1 – 2 of 2) sorted by relevance
106 const TensorShapeProto& reduction_indices_shape = in Optimize() local108 if (NumCoefficients(reduction_indices_shape) == 1) { in Optimize()
756 PartialTensorShape reduction_indices_shape(reduction_indices_prop.shape()); in MaterializeReductionIndices() local757 const int num_reduction_indices = reduction_indices_shape.num_elements(); in MaterializeReductionIndices()2737 const TensorShapeProto& reduction_indices_shape = in IsReductionWithConstantIndices() local2739 *indices_is_empty = TensorShape(reduction_indices_shape).num_elements() == 0; in IsReductionWithConstantIndices()