Searched refs:prefix_op (Results 1 – 3 of 3) sorted by relevance
180 BlockPrefixCallbackOp<T, Op> prefix_op(IdentityValue<T, Op>()(), op); in scan_kernel()214 .ExclusiveScan(thread_data, thread_data, op, prefix_op); in scan_kernel()217 .InclusiveScan(thread_data, thread_data, op, prefix_op); in scan_kernel()
449 OpCode prefix_op () const { return prefix_op_; } in prefix_op() function930 if (str.prefix_op () != OpCode_Invalid) in encode_str()931 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()