Searched refs:prefix_op (Results 1 – 3 of 3) sorted by relevance
206 BlockPrefixCallbackOp<T, Op> prefix_op(IdentityValue<T, Op>()(), op); in __launch_bounds__()240 .ExclusiveScan(thread_data, thread_data, op, prefix_op); in __launch_bounds__()243 .InclusiveScan(thread_data, thread_data, op, prefix_op); in __launch_bounds__()
450 op_code_t prefix_op () const { return prefix_op_; } in prefix_op() function959 if (str.prefix_op () != OpCode_Invalid) in encode_str()960 encoder.encode_op (str.prefix_op ()); in encode_str()
253 Node* prefix_op = in_edges[0]->src(); in AddRestoreVariableSubgraphs() local296 .Input(prefix_op) in AddRestoreVariableSubgraphs()