Searched refs:TensorListGetItemOp (Results 1 – 5 of 5) sorted by relevance
339 class TensorListGetItemOp : public XlaOpKernel { class341 explicit TensorListGetItemOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TensorListGetItemOp() function in tensorflow::__anond80ffd040111::TensorListGetItemOp372 TF_DISALLOW_COPY_AND_ASSIGN(TensorListGetItemOp);375 REGISTER_XLA_OP(Name("TensorListGetItem"), TensorListGetItemOp);
711 : public OpConversionPattern<TF::TensorListGetItemOp> {715 TF::TensorListGetItemOp op, ArrayRef<Value> operands, in matchAndRewrite()818 if (llvm::isa<TF::TensorListGetItemOp>(op) || in GetTensorListArgumentsFromWhileOp()991 TF::TensorListGetItemOp, TF::TensorListLengthOp, in runOnOperation()
136 GET_OPERATION_NAME(TF::TensorListGetItemOp), in AddRewrittenCompositeOps()
688 TF::TensorListGetItemOp get_item, in HandleTensorListGetItemOp()844 } else if (auto get_item = llvm::dyn_cast<TF::TensorListGetItemOp>(&op)) { in DecomposeTensorListOpsInternal()
271 TensorListLengthOp, TensorListGetItemOp>(use.getOwner())) in CanInferTensorListElementType()