Searched refs:TensorListFromTensorOp (Results 1 – 5 of 5) sorted by relevance
574 class TensorListFromTensorOp : public XlaOpKernel { class576 explicit TensorListFromTensorOp(OpKernelConstruction* ctx) in TensorListFromTensorOp() function in tensorflow::__anond80ffd040111::TensorListFromTensorOp591 TF_DISALLOW_COPY_AND_ASSIGN(TensorListFromTensorOp);596 TensorListFromTensorOp);
133 GET_OPERATION_NAME(TF::TensorListFromTensorOp), in AddRewrittenCompositeOps()
140 isa<TensorListFromTensorOp>(op); in IsTensorListInitOp()149 if (auto tl_from_tensor = dyn_cast<TensorListFromTensorOp>(op)) in GetElementShapeOperand()761 if (auto tl_from_tensor = dyn_cast<TensorListFromTensorOp>(op)) { in InferShapeForTensorListInitOps()
618 TF::TensorListFromTensorOp list, in HandleTensorListFromTensorOp()832 } else if (auto list = llvm::dyn_cast<TF::TensorListFromTensorOp>(&op)) { in DecomposeTensorListOpsInternal()
990 target.addIllegalOp<TF::EmptyTensorListOp, TF::TensorListFromTensorOp, in runOnOperation()