Searched refs:TensorListElementShapeOp (Results 1 – 5 of 5) sorted by relevance
282 class TensorListElementShapeOp : public XlaOpKernel { class284 explicit TensorListElementShapeOp(OpKernelConstruction* ctx) in TensorListElementShapeOp() function in tensorflow::__anond80ffd040111::TensorListElementShapeOp333 TF_DISALLOW_COPY_AND_ASSIGN(TensorListElementShapeOp);337 TensorListElementShapeOp);
139 GET_OPERATION_NAME(TF::TensorListElementShapeOp), in AddRewrittenCompositeOps()
757 TF::TensorListElementShapeOp elem_shape, in HandleTensorListElementShapeOp()860 llvm::dyn_cast<TF::TensorListElementShapeOp>(&op)) { in DecomposeTensorListOpsInternal()
264 dyn_cast<TensorListElementShapeOp>(use.getOwner())) { in CanInferTensorListElementType()
2193 OpFoldResult TensorListElementShapeOp::fold(ArrayRef<Attribute> operands) { in fold()