Searched defs:final_op (Results 1 – 5 of 5) sorted by relevance
156 final_op=None, argument229 final_op=None, argument
24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
103 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv() local
123 std::unique_ptr<OpKernel> final_op; // reduction only member
209 } final_op; member