Searched refs:unpacked_indices (Results 1 – 1 of 1) sorted by relevance
242 auto unpacked_indices = rewriter->create<TF::UnpackOp>( in UnpackTensorAlongZeroDim() local245 return unpacked_indices; in UnpackTensorAlongZeroDim()5194 auto unpacked_indices = rewriter.create<TF::UnpackOp>( in matchAndRewrite() local5221 llvm::zip(unpacked_indices.output(), split_updates.output())) { in matchAndRewrite()5248 auto unpacked_indices = rewriter.create<TF::UnpackOp>( in matchAndRewrite() local5252 op, op.getType(), op.input(), op.update(), unpacked_indices.output()); in matchAndRewrite()