Searched refs:TensorListLengthOp (Results 1 – 5 of 5) sorted by relevance
87 class TensorListLengthOp : public XlaOpKernel { class89 explicit TensorListLengthOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in TensorListLengthOp() function in tensorflow::__anond80ffd040111::TensorListLengthOp102 TF_DISALLOW_COPY_AND_ASSIGN(TensorListLengthOp);105 REGISTER_XLA_OP(Name("TensorListLength").IsMetadataOp(), TensorListLengthOp);
574 auto input_size = rewriter.create<TF::TensorListLengthOp>( in matchAndRewrite()726 : public OpConversionPattern<TF::TensorListLengthOp> {730 TF::TensorListLengthOp op, ArrayRef<Value> operands, in matchAndRewrite()819 llvm::isa<TF::TensorListLengthOp>(op) || in GetTensorListArgumentsFromWhileOp()991 TF::TensorListGetItemOp, TF::TensorListLengthOp, in runOnOperation()
138 GET_OPERATION_NAME(TF::TensorListLengthOp), in AddRewrittenCompositeOps()
728 TF::TensorListLengthOp length, in HandleTensorListLengthOp()852 } else if (auto length = llvm::dyn_cast<TF::TensorListLengthOp>(&op)) { in DecomposeTensorListOpsInternal()
271 TensorListLengthOp, TensorListGetItemOp>(use.getOwner())) in CanInferTensorListElementType()