Searched refs:dim_to_reduce (Results 1 – 2 of 2) sorted by relevance
6 %dim_to_reduce = "tf.Const"() {value = dense<[1]> : tensor<1xi32>}8 %0 = "tf.Sum"(%input, %dim_to_reduce) {keep_dims = false}25 %dim_to_reduce = "tf.Const"() {value = dense<[0]> : tensor<1xi32>}27 %0 = "tf.Sum"(%input, %dim_to_reduce) {keep_dims = false}44 %dim_to_reduce = "tf.Const"() {value = dense<[1]> : tensor<1xi32>}46 %0 = "tf.Mean"(%input, %dim_to_reduce) {keep_dims = false}66 %dim_to_reduce = "tf.Const"() {value = dense<[0]> : tensor<1xi32>}68 %0 = "tf.Sum"(%input, %dim_to_reduce) {keep_dims = false}
2262 for (int64_t dim_to_reduce : dimensions_to_reduce) { in InferReduceShape() local2263 if (!dimensions_to_reduce_set.insert(dim_to_reduce).second) { in InferReduceShape()2265 dim_to_reduce); in InferReduceShape()