Searched refs:TensorListReserveOp (Results 1 – 6 of 6) sorted by relevance
134 class TensorListReserveOp : public XlaOpKernel { class136 explicit TensorListReserveOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListReserveOp() function in tensorflow::__anond80ffd040111::TensorListReserveOp201 TF_DISALLOW_COPY_AND_ASSIGN(TensorListReserveOp);207 TensorListReserveOp);
482 : public ConvertTensorListInitOp<TF::TensorListReserveOp> {486 Value GetNumElements(TF::TensorListReserveOp op, ArrayRef<Value> operands, in GetNumElements()662 auto extended_part = rewriter->create<TF::TensorListReserveOp>( in CreateCondTrueBranch()821 llvm::isa<TF::TensorListReserveOp>(op) || in GetTensorListArgumentsFromWhileOp()992 TF::TensorListPushBackOp, TF::TensorListReserveOp, in runOnOperation()
132 GET_OPERATION_NAME(TF::TensorListReserveOp), in AddRewrittenCompositeOps()
590 TF::TensorListReserveOp list, in HandleTensorListReserveOp()828 } else if (auto list = llvm::dyn_cast<TF::TensorListReserveOp>(&op)) { in DecomposeTensorListOpsInternal()
139 return isa<TensorListReserveOp>(op) || isa<EmptyTensorListOp>(op) || in IsTensorListInitOp()147 if (auto tl_reserve = dyn_cast<TensorListReserveOp>(op)) in GetElementShapeOperand()
2177 static LogicalResult Verify(TensorListReserveOp op) { in Verify()