Searched refs:tl_from_tensor (Results 1 – 1 of 1) sorted by relevance
239 if (auto tl_from_tensor = dyn_cast<TensorListFromTensorOp>(op)) in GetElementShapeOperand() local240 return tl_from_tensor.element_shape(); in GetElementShapeOperand()995 if (auto tl_from_tensor = dyn_cast<TensorListFromTensorOp>(op)) { in InferShapeForTensorListInitOps() local998 element_type = DropFirstDimension(tl_from_tensor.tensor().getType()); in InferShapeForTensorListInitOps()