Searched refs:reduce_dims_op (Results 1 – 1 of 1) sorted by relevance
598 auto reduce_dims_op = builder->create<ConstOp>(loc, reduce_dims); in ApplyReduction() local599 return builder->create<TF::SumOp>(loc, input, reduce_dims_op, in ApplyReduction()